Searched refs:MatchingArraySize (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | QuantizedLSTM.cpp | 80 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 D | types.h | 249 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 D | legacy_reference_ops.h | 739 /* 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()
|