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 D | conv_2d_builder.cc | 131 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 D | conv_2d_builder.h | 74 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 D | quantize_op.cc | 167 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 D | quantized_concat_op.cc | 229 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 D | quantized_batch_norm_op.cc | 280 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 D | mkl_quantize_op.cc | 293 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 D | mkl_concat_op.cc | 232 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()
|