| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 26 void InitTensor(const std::vector<int>& shape, TfLiteAllocationType alloc_type, in InitTensor() function
47 InitTensor(shape, kTfLiteArenaRw, 1.0f, &tensor); in TEST()
73 InitTensor(shape, kTfLiteDynamic, 1.0f, &tensor); in TEST()
100 InitTensor(shape_a, kTfLiteDynamic, 1.0, &tensor_a); in TEST()
101 InitTensor(shape_b, kTfLiteDynamic, 4.0, &tensor_b); in TEST()
141 InitTensor(shape_a, kTfLiteDynamic, 1.0, &tensor_a); in TEST()
142 InitTensor(shape_b, kTfLiteDynamic, 4.0, &tensor_b); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device_test.cc | 27 void InitTensor(Tensor* tensor, float value) { in InitTensor() function
52 InitTensor(&input, 1); in TEST()
53 InitTensor(&output, 0); in TEST()
|
| D | scoped_allocator_mgr_test.cc | 30 void InitTensor() { in InitTensor() function in tensorflow::__anon9b7b749e0111::ScopedAllocatorMgrTest
49 InitTensor(); in PrepScopedAllocatorMgr()
110 InitTensor(); in TEST_F()
184 InitTensor(); in TEST_F()
|
| D | permuter_test.cc | 256 instance->InitTensor( in RunTest()
350 void InitTensor(DataType dtype, const TensorShape& shape, in InitTensor() function in tensorflow::__anon153c086f0111::PermuterTest::DeviceInstance
|
| D | ring_gatherer_test.cc | 282 instance->InitTensor(dtype, TensorShape({tensor_len}), in RunTest()
429 void InitTensor(DataType dtype, const TensorShape& shape, in InitTensor() function in tensorflow::RingGathererTest::DeviceInstance
|
| D | ring_reducer_test.cc | 304 instance->InitTensor( in RunTest()
455 void InitTensor(DataType dtype, const TensorShape& shape, in InitTensor() function in tensorflow::RingReducerTest::DeviceInstance
|
| D | hierarchical_tree_broadcaster_test.cc | 429 instance->InitTensor( in RunTest()
610 void InitTensor(DataType dtype, const TensorShape& shape, in InitTensor() function in tensorflow::__anonb5878a120111::HierarchicalTreeBroadcasterTest::DeviceInstance
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 203 instance->InitTensor(DT_FLOAT, TensorShape({input_length}), in RunTest()
267 void InitTensor(DataType dtype, const TensorShape& shape, in InitTensor() function in tensorflow::NcclTestBase::DeviceInstance
|