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()
|