experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
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
…
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