Home
last modified time | relevance | path

Searched defs:input_dims_size (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
H A Dlayer_norm.cc121 bool ValidIndex(const int* input_dims, const int input_dims_size, in ValidIndex()
143 int GetOffset(const int* input_dims, const int input_dims_size, in GetOffset()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dmatrix_diag.cc38 int input_dims_size = input_dims->size; in Prepare() local
H A Dmatrix_set_diag.cc39 int input_dims_size = input_dims->size; in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_api_experimental.cc156 const int* input_dims, int32_t input_dims_size) { in TfLiteSignatureRunnerResizeInputTensor()
H A Dc_api.cc173 int32_t input_dims_size) { in TfLiteInterpreterResizeInputTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
H A DInterpreter.cs240 int input_dims_size); in TfLiteInterpreterResizeInputTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Ddelegate_test_util.cc320 const int input_dims_size = input1->dims->size; in FakeFusedRegistration() local