Home
last modified time | relevance | path

Searched refs:BuildTfLiteIntArray (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dquantization_util_test.cc40 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 Dutil.h75 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray(
H A Dutil.cc50 std::unique_ptr<TfLiteIntArray, TfLiteIntArrayDeleter> BuildTfLiteIntArray( in BuildTfLiteIntArray() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/utils/
H A Dsimple_delegate.cc112 BuildTfLiteIntArray(supported_nodes).get(), base_delegate); in DelegatePrepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/
H A Dcoreml_delegate.mm255 context, GetCoreMlKernelRegistration(), BuildTfLiteIntArray(delegated_nodes).get(), delegate);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc6566 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()