Lines Matching refs:GetTensorShape

139   type::AveragePool(op_params, GetTensorShape(input),                    \  in AverageEvalFloat()
140                     GetTensorData<float>(input), GetTensorShape(output), \  in AverageEvalFloat()
169   type::AveragePool(op_params, GetTensorShape(input),                      \  in AverageEvalQuantizedUint8()
170                     GetTensorData<uint8_t>(input), GetTensorShape(output), \  in AverageEvalQuantizedUint8()
197       op_params, GetTensorShape(input), GetTensorData<int8_t>(input),  in AverageEvalQuantizedInt8()
198       GetTensorShape(output), GetTensorData<int8_t>(output));  in AverageEvalQuantizedInt8()
218   type::MaxPool(op_params, GetTensorShape(input), GetTensorData<float>(input), \  in MaxEvalFloat()
219                 GetTensorShape(output), GetTensorData<float>(output))  in MaxEvalFloat()
246   type::MaxPool(op_params, GetTensorShape(input),                      \  in MaxEvalQuantizedUInt8()
247                 GetTensorData<uint8_t>(input), GetTensorShape(output), \  in MaxEvalQuantizedUInt8()
275   type::MaxPool(op_params, GetTensorShape(input),                     \  in MaxEvalQuantizedInt8()
276                 GetTensorData<int8_t>(input), GetTensorShape(output), \  in MaxEvalQuantizedInt8()
299   type::L2Pool(op_params, GetTensorShape(input), GetTensorData<float>(input), \  in L2EvalFloat()
300                GetTensorShape(output), GetTensorData<float>(output))  in L2EvalFloat()