Home
last modified time | relevance | path

Searched refs:tensor_idx (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator_test.cc104 for (int tensor_idx = 0; tensor_idx < 4; tensor_idx++) { in TEST() local
105 EXPECT_NEAR(stats.at(tensor_idx).min, tensor_idx + 1, eps); in TEST()
106 EXPECT_NEAR(stats.at(tensor_idx).max, tensor_idx + 1, eps); in TEST()
163 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
164 EXPECT_NEAR(stats.at(tensor_idx).min, expected_values[tensor_idx], eps); in TEST()
165 EXPECT_NEAR(stats.at(tensor_idx).max, expected_values[tensor_idx], eps); in TEST()
180 for (int tensor_idx = 1; tensor_idx < 5; tensor_idx++) { in TEST() local
181 EXPECT_NEAR(stats.at(tensor_idx).min, expected_values[tensor_idx], eps); in TEST()
182 EXPECT_NEAR(stats.at(tensor_idx).max, expected_values[tensor_idx], eps); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc51 int32_t tensor_idx) { in GetTensorConsumers() argument
61 if (op->inputs[i] == tensor_idx) { in GetTensorConsumers()
130 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local
132 if (tensor_idx == -1) { in CheckAllOpInputsQuantized()
137 TensorT* tensor = subgraph->tensors[tensor_idx].get(); in CheckAllOpInputsQuantized()
157 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local
158 if (tensor_idx == -1) { in InsertQuantizableInputTensorsFromOperator()
164 TensorT* tensor = subgraph->tensors[tensor_idx].get(); in InsertQuantizableInputTensorsFromOperator()
188 tensor_map->insert({tensor_idx, tensor}); in InsertQuantizableInputTensorsFromOperator()
288 const int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsInternal() local
[all …]
Dsubgraph_quantizer.h60 int32_t tensor_idx);
63 bool IsSubgraphInput(int32_t tensor_idx) const;
Dsubgraph_quantizer.cc197 BuiltinOperator op_code, int32_t tensor_idx) { in AsymmetricQuantizeTensor() argument
198 TensorT* tensor = subgraph_->tensors[tensor_idx].get(); in AsymmetricQuantizeTensor()
211 tensor_idx, EnumNameBuiltinOperator(op_code)); in AsymmetricQuantizeTensor()
354 bool SubgraphQuantizer::IsSubgraphInput(int32_t tensor_idx) const { in IsSubgraphInput()
356 tensor_idx) != subgraph_->inputs.end(); in IsSubgraphInput()
Dquantize_weights_test.cc73 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput() argument
78 if (subgraph->inputs()->Get(i) == tensor_idx) { in IsModelInputOrOutput()
83 if (subgraph->outputs()->Get(i) == tensor_idx) { in IsModelInputOrOutput()
93 uint32_t tensor_idx, in GetProducerOpCode() argument
99 if (op->outputs()->Get(i) == tensor_idx) { in GetProducerOpCode()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h87 const Tensor* tensor_idx = std::get<0>(*tensor); in ValidateShape() local
112 tensor_idx->dims() > 0) { in ValidateShape()
113 for (int64 i = 0; i < tensor_idx->dim_size(0); i++) { in ValidateShape()
114 if (tensor_idx->vec<int64>()(i) >= shape_.dim_size(0)) { in ValidateShape()
117 "; index is ", tensor_idx->vec<int64>()(i), " exceeded ", in ValidateShape()
Ddepthwise_conv_op_gpu.h233 const int tensor_idx = thread_pix * in_depth + thread_depth; in DepthwiseConv2dGPUKernelNHWCSmall() local
257 const int filter_offset = tensor_idx + start_channel; in DepthwiseConv2dGPUKernelNHWCSmall()
516 const int tensor_idx = thread_depth * in_pixels + thread_pix; in DepthwiseConv2dGPUKernelNCHWSmall() local
542 const int inout_offset = channel * in_pixels + tensor_idx; in DepthwiseConv2dGPUKernelNCHWSmall()
1229 const int tensor_idx = thread_pix * in_depth + thread_depth;
1252 const int filter_offset = tensor_idx + start_channel;
1500 const int tensor_idx = thread_depth * in_pixels + thread_pix;
1520 const int inout_offset = channel * in_pixels + tensor_idx;
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc207 for (const int tensor_idx : *subgraph.inputs()) { in VerifySubGraphConsistency()
208 subgraph_input_tensors.insert(tensor_idx); in VerifySubGraphConsistency()
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py850 tensor_idx = self._cache_idx_to_tensor_idx[cache_idx]
851 line = '%d %d\n'%(cache_idx, tensor_idx)
1162 tensor_idx = self._tensorname_idx_map[tensor_name]
1165 self._cache_idx_to_tensor_idx.append(tensor_idx)