| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantize_and_dequantize_op.cc | 86 Tensor input_max_tensor; in Compute() local
91 input_max_tensor = ctx->input(2); in Compute()
94 auto max_val = input_max_tensor.scalar<T>()(); in Compute()
106 ctx, input_max_tensor.dim_size(0) == depth, in Compute()
108 input_max_tensor.dim_size(0), " expected ", depth, in Compute()
110 input_max_tensor.shape())); in Compute()
117 range_shape, &input_max_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()
179 const Tensor& input_max_tensor = ctx->input(3); 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 ()()
39 input_max_tensor, round_mode, narrow_range, output); in operator ()()
47 Tensor* input_min_tensor, Tensor* input_max_tensor, in operator ()()
52 input_max_tensor, round_mode, narrow_range, output); in operator ()()
61 typename TTypes<T>::ConstScalar input_max_tensor, in operator ()()
66 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
77 const Tensor* input_max_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,
79 const Tensor* input_max_tensor,
184 Tensor* input_min_tensor, Tensor* input_max_tensor, in Compute()
190 auto input_max = input_max_tensor->scalar<T>(); in Compute()
199 max_range = input_max_tensor->scalar<T>()(); in Compute()
225 Tensor* input_min_tensor, Tensor* input_max_tensor, in Compute()
231 auto input_max = input_max_tensor->vec<T>(); in Compute()
247 std::memcpy(max_range.data(), input_max_tensor->vec<T>().data(), in Compute()
296 const Tensor* input_max_tensor, in Compute()
[all …]
|
| H A D | dequantize_op.cc | 95 const Tensor& input_max_tensor = ctx->input(2); in Compute() local
112 OP_REQUIRES(ctx, input_max_tensor.NumElements() == num_slices, in Compute()
116 axis_, "), got ", input_max_tensor.NumElements(), in Compute()
125 const float max_range = input_max_tensor.flat<float>()(0); in Compute()
144 auto max_ranges = input_max_tensor.vec<float>(); in Compute()
|
| H A D | quantized_batch_norm_op.cc | 180 const auto& input_max_tensor = context->input(2); in Compute() local
181 OP_REQUIRES(context, input_max_tensor.NumElements() == 1, in Compute()
183 const float input_max = input_max_tensor.flat<float>()(0); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-070.md | 35 const float max_range = input_max_tensor.flat<float>()(0);
40 auto max_ranges = input_max_tensor.vec<float>();
|
| H A D | tfsa-2021-031.md | 36 const Tensor* input_max_tensor,
42 auto input_max = input_max_tensor->vec<T>();
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | quantize_training.cc | 514 Tensor input_max_tensor(DT_FLOAT, TensorShape()); in MakeInputMinMax() local
515 input_max_tensor.flat<float>()(0) = edge.input_max; in MakeInputMinMax()
519 .Attr("value", input_max_tensor) in MakeInputMinMax()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | quantize_nodes_test.cc | 934 Tensor input_max_tensor(DT_FLOAT, TensorShape({})); in TestMergeAdjacentRequantizes() local
935 test::FillValues<float>(&input_max_tensor, {255.0f}); in TestMergeAdjacentRequantizes()
937 Input::Initializer(input_max_tensor)); in TestMergeAdjacentRequantizes()
|