Searched defs:tensor_idx (Results 1 – 6 of 6) sorted by relevance
104 for (int tensor_idx = 0; tensor_idx < 4; tensor_idx++) { in TEST() local163 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local180 for (int tensor_idx = 1; tensor_idx < 5; tensor_idx++) { in TEST() local
51 int32_t tensor_idx) { in GetTensorConsumers()130 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local157 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local288 const int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsInternal() local
73 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput()93 uint32_t tensor_idx, in GetProducerOpCode()
197 BuiltinOperator op_code, int32_t tensor_idx) { in AsymmetricQuantizeTensor()
233 const int tensor_idx = thread_pix * in_depth + thread_depth; in DepthwiseConv2dGPUKernelNHWCSmall() local516 const int tensor_idx = thread_depth * in_pixels + thread_pix; in DepthwiseConv2dGPUKernelNCHWSmall() local
87 const Tensor* tensor_idx = std::get<0>(*tensor); in ValidateShape() local