experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
rei/fork-r2.3
STT-tensorflow/tensorflow/lite/delegates/flex
History
Xiao Yu fe6e64b098 Refactor eager placement logic into three util methods: 
- MaybePinSmallOpsToCpu
- MaybePinToResourceDevice
- MaybePinToCustomDevice

We are going to reuse MaybePinSmallOpsToCpu in TFRT but not the other two. Because TFRT doesn't have native Resource neither Custom Device.

PiperOrigin-RevId: 317766813
Change-Id: I43241b5786120ddf39dc4bfff6071239afdfd785
2020-06-22 18:05:46 -07:00
..
java/src/main Selective registration for flex ops 2020-06-07 20:02:02 -07:00
buffer_map_test.cc Fix tensor initialization in unit test. 2019-11-20 18:31:24 -08:00
buffer_map.cc Use GetStringCount(tensor) and GetString(tensor, i) instead of the raw pointer overloads where applicable. 2019-10-08 20:21:10 -07:00
buffer_map.h Rename lite/c/c_api_internal.h to lite/c/common.h 2019-11-25 21:49:25 -08:00
BUILD Disable tsan on InterpreterFlexTest and SelectiveBuiltInterpreterFlexTest 2020-06-18 19:35:34 -07:00
build_def.bzl Add rule to generate selective-built android library for flex delegate 2020-06-17 23:23:08 -07:00
delegate_data_test.cc Rename lite/c/c_api_internal.h to lite/c/common.h 2019-11-25 21:49:25 -08:00
delegate_data.cc Add lazy_remote_inputs_copy to TFE_ContextOptions to control lazy remote tensor copy. Disable it by default. 2019-11-07 18:35:44 -08:00
delegate_data.h Refcount the EagerContext. 2019-05-01 13:42:06 -07:00
delegate_test.cc 1. Adapt flex delegate to the newly created SimpleDelegate APIs. 2020-06-01 16:28:37 -07:00
delegate.cc Enable flex delegate on tensorflow.lite.Interpreter Python package 2020-06-18 00:01:30 -07:00
delegate.h Small changes to SimpleDelegate APIs: 2020-06-02 01:37:30 -07:00
kernel_test.cc 1. Adapt flex delegate to the newly created SimpleDelegate APIs. 2020-06-01 16:28:37 -07:00
kernel.cc Refactor eager placement logic into three util methods: 2020-06-22 18:05:46 -07:00
kernel.h Small changes to SimpleDelegate APIs: 2020-06-02 01:37:30 -07:00
test_util.cc Change comment of the external repo #include file origin. 2020-03-23 10:42:53 -07:00
test_util.h Unittests to demonstrate three bugs in flex mode: 2019-01-07 14:56:06 -08:00
util_test.cc Add Mbed CI testing support 2019-09-05 13:53:45 -07:00
util.cc Add float64 tensor support in TFLite 2020-04-03 23:20:03 -07:00
util.h Rename lite/c/c_api_internal.h to lite/c/common.h 2019-11-25 21:49:25 -08:00
whitelisted_flex_ops.cc Support StringToHashBucketFast, Real, Imag ops in TFLite flex delegate 2020-06-03 19:09:44 -07:00
whitelisted_flex_ops.h Move flex ops whitelist to a library outside toco 2019-05-23 22:40:14 -07:00