Searched refs:BuildTfLiteIntArray (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
H A D | quantization_util_test.cc | 40 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray( in BuildTfLiteIntArray() function 149 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() 174 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() 199 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() 221 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/ |
H A D | util.h | 75 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray(
|
H A D | util.cc | 50 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray( in BuildTfLiteIntArray() function
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/utils/ |
H A D | simple_delegate.cc | 112 BuildTfLiteIntArray(supported_nodes).get(), base_delegate); in DelegatePrepare()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/ |
H A D | coreml_delegate.mm | 255 context, GetCoreMlKernelRegistration(), BuildTfLiteIntArray(delegated_nodes).get(), delegate);
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
H A D | nnapi_delegate.cc | 6566 auto supported_nodes_int_array = BuildTfLiteIntArray(supported_nodes); in GetNodesSupportedByAccelerator() 6600 BuildTfLiteIntArray(*device_supported_nodes); in GetNodesSupportedByAccelerator() 6888 auto supported_nodes_int_array = BuildTfLiteIntArray(supported_nodes); in DoPrepare() 6930 auto nodes_to_delegate_int_array = BuildTfLiteIntArray(nodes_to_delegate); in DoPrepare()
|