Home
last modified time | relevance | path

Searched refs:MatchingArraySize (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQuantizedLSTM.cpp80 TFLITE_CHECK_EQ(MatchingArraySize(bias_dims, 1, bias_dims, 2, bias_dims, 3), in quantizedLstmStep()
83 MatchingArraySize(weights_dims, 1, bias_dims, 0); in quantizedLstmStep()
86 MatchingArraySize(prevCellState_dims, 0, prev_activ_dims, 0, in quantizedLstmStep()
91 MatchingArraySize(weights_dims, 1, activ_temp_dims, 0); in quantizedLstmStep()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtypes.h249 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
256 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
259 return MatchingArraySize(array1, index1, args...); in MatchingArraySize()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlegacy_reference_ops.h739 /* batches = */ MatchingArraySize(*output_dims[i], 3, input_dims, 3); in TensorFlowSplit()
740 /* height = */ MatchingArraySize(*output_dims[i], 2, input_dims, 2); in TensorFlowSplit()
741 /* width = */ MatchingArraySize(*output_dims[i], 1, input_dims, 1); in TensorFlowSplit()