| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantize_and_dequantize_op.cc | 85 Tensor input_min_tensor; in Compute() local
90 input_min_tensor = ctx->input(1); in Compute()
93 auto min_val = input_min_tensor.scalar<T>()(); in Compute()
100 ctx, input_min_tensor.dim_size(0) == depth, in Compute()
102 input_min_tensor.dim_size(0), " expected ", depth, in Compute()
104 input_min_tensor.shape())); in Compute()
115 range_shape, &input_min_tensor)); in Compute()
123 range_given_, &input_min_tensor, &input_max_tensor, round_mode_, in Compute()
129 num_bits_, range_given_, &input_min_tensor, &input_max_tensor, in Compute()
173 const Tensor& input_min_tensor = ctx->input(2); in Compute() local
[all …]
|
| H A D | quantize_and_dequantize_op_gpu.cu.cc | 34 Tensor* input_min_tensor, Tensor* input_max_tensor, in operator ()()
38 d, input, signed_input, num_bits, range_given, input_min_tensor, in operator ()()
47 Tensor* input_min_tensor, Tensor* input_max_tensor, in operator ()()
51 d, input, signed_input, num_bits, range_given, input_min_tensor, in operator ()()
60 typename TTypes<T>::ConstScalar input_min_tensor, in operator ()()
66 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
76 const Tensor* input_min_tensor, in operator ()()
82 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
|
| H A D | quantize_and_dequantize_op.h | 49 Tensor* input_min_tensor, Tensor* input_max_tensor,
58 Tensor* input_min_tensor, Tensor* input_max_tensor,
78 const Tensor* input_min_tensor,
184 Tensor* input_min_tensor, Tensor* input_max_tensor, in Compute()
189 auto input_min = input_min_tensor->scalar<T>(); in Compute()
198 min_range = input_min_tensor->scalar<T>()(); in Compute()
225 Tensor* input_min_tensor, Tensor* input_max_tensor, in Compute()
230 auto input_min = input_min_tensor->vec<T>(); in Compute()
245 std::memcpy(min_range.data(), input_min_tensor->vec<T>().data(), in Compute()
295 const Tensor* input_min_tensor, in Compute()
[all …]
|
| H A D | dequantize_op.cc | 94 const Tensor& input_min_tensor = ctx->input(1); in Compute() local
106 OP_REQUIRES(ctx, input_min_tensor.NumElements() == num_slices, in Compute()
110 axis_, "), got ", input_min_tensor.NumElements(), in Compute()
124 const float min_range = input_min_tensor.flat<float>()(0); in Compute()
143 auto min_ranges = input_min_tensor.vec<float>(); in Compute()
|
| H A D | quantized_batch_norm_op.cc | 176 const auto& input_min_tensor = context->input(1); in Compute() local
177 OP_REQUIRES(context, input_min_tensor.NumElements() == 1, in Compute()
179 const float input_min = input_min_tensor.flat<float>()(0); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-070.md | 34 const float min_range = input_min_tensor.flat<float>()(0);
39 auto min_ranges = input_min_tensor.vec<float>();
|
| H A D | tfsa-2021-031.md | 35 const Tensor* input_min_tensor,
41 auto input_min = input_min_tensor->vec<T>();
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | quantize_training.cc | 507 Tensor input_min_tensor(DT_FLOAT, TensorShape()); in MakeInputMinMax() local
508 input_min_tensor.flat<float>()(0) = edge.input_min; in MakeInputMinMax()
512 .Attr("value", input_min_tensor) in MakeInputMinMax()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | quantize_nodes_test.cc | 929 Tensor input_min_tensor(DT_FLOAT, TensorShape({})); in TestMergeAdjacentRequantizes() local
930 test::FillValues<float>(&input_min_tensor, {0.0f}); in TestMergeAdjacentRequantizes()
932 Input::Initializer(input_min_tensor)); in TestMergeAdjacentRequantizes()
|