Home
last modified time | relevance | path

Searched refs:output_max_tensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_builder.cc131 TensorID* output_min_tensor, TensorID* output_max_tensor) { in BuildDilatedDwConv() argument
233 *output_max_tensor = in BuildDilatedDwConv()
245 TensorID* output_min_tensor, TensorID* output_max_tensor) { in BuildSplittedDwConv() argument
325 *output_max_tensor = concat->AddOutput(sizeof(float), 4, kScalarShape); in BuildSplittedDwConv()
334 TensorID* output_max_tensor) { in BuildStandardConv() argument
368 *output_max_tensor = AddOutput(sizeof(float), 4, kScalarShape); in BuildStandardConv()
477 TensorID output_tensor, output_min_tensor, output_max_tensor; in PopulateSubGraph() local
483 &output_max_tensor); in PopulateSubGraph()
488 &output_tensor, &output_min_tensor, &output_max_tensor); in PopulateSubGraph()
493 &output_tensor, &output_min_tensor, &output_max_tensor); in PopulateSubGraph()
[all …]
H A Dconv_2d_builder.h74 TensorID* output_max_tensor);
84 TensorID* output_max_tensor);
91 TensorID* output_min_tensor, TensorID* output_max_tensor);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_op.cc167 Tensor* output_max_tensor = nullptr; in Compute() local
171 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, {}, &output_max_tensor)); in Compute()
175 output_min_tensor, output_max_tensor); in Compute()
185 ctx->allocate_output(2, minmax_shape, &output_max_tensor)); in Compute()
205 &output_max_tensor->flat<float>()(i)); in Compute()
212 Tensor* output_max_tensor) { in QuantizeTensor() argument
245 output_max_tensor->flat<float>()(0) = max_range; in QuantizeTensor()
251 &output_max_tensor->flat<float>()(0)); in QuantizeTensor()
H A Dquantized_concat_op.cc229 Tensor* output_max_tensor = nullptr; in Compute() local
231 context->allocate_output(2, {}, &output_max_tensor)); in Compute()
232 output_max_tensor->flat<float>()(0) = output_max; in Compute()
H A Dquantized_batch_norm_op.cc280 Tensor* output_max_tensor = nullptr; in Compute() local
282 context->allocate_output(2, {}, &output_max_tensor)); in Compute()
283 output_max_tensor->flat<float>()(0) = output_max; in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_quantize_op.cc293 Tensor* output_max_tensor = nullptr; in ComputeScalar() local
294 AllocateOutputSetMklShape(ctx, 2, &output_max_tensor, max_tf_shape, in ComputeScalar()
308 output_max_tensor->flat<float>()(0) = max_range; in ComputeScalar()
443 Tensor* output_max_tensor = nullptr; in Compute() local
444 AllocateOutputSetMklShape(ctx, 2, &output_max_tensor, max_tf_shape, in Compute()
492 output_max_tensor->flat<float>()(0) = max_range; in Compute()
H A Dmkl_concat_op.cc232 Tensor* output_max_tensor = nullptr; in Compute() local
233 OP_REQUIRES_OK(c, c->allocate_output(2, {}, &output_max_tensor)); in Compute()
234 output_max_tensor->flat<float>()(0) = output_max; in Compute()