Searched refs:input_min_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantize_and_dequantize_op.cc | 75 Tensor input_min_tensor; in Compute() local
78 input_min_tensor = ctx->input(1); in Compute()
80 auto min_val = input_min_tensor.scalar<T>()(); in Compute()
87 TensorShape(), &input_min_tensor)); in Compute()
94 range_given_, &input_min_tensor, &input_max_tensor, round_mode_, in Compute()
136 Tensor input_min_tensor; in Compute() local
139 input_min_tensor = ctx->input(1); in Compute()
141 auto min_val = input_min_tensor.scalar<T>()(); in Compute()
148 TensorShape(), &input_min_tensor)); in Compute()
155 range_given_, &input_min_tensor, &input_max_tensor, ROUND_HALF_TO_EVEN, in Compute()
[all …]
|
| D | quantize_and_dequantize_op.h | 47 Tensor* input_min_tensor, Tensor* input_max_tensor,
109 Tensor* input_min_tensor, Tensor* input_max_tensor, in Compute()
114 auto input_min = input_min_tensor->scalar<T>(); in Compute()
123 min_range = input_min_tensor->scalar<T>()(); in Compute()
|
| D | quantize_and_dequantize_op_gpu.cu.cc | 34 Tensor* input_min_tensor, Tensor* input_max_tensor, in operator ()()
37 d, input, signed_input, num_bits, range_given, input_min_tensor, in operator ()()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 509 Tensor input_min_tensor(DT_FLOAT, TensorShape()); in MakeInputMinMax() local
510 input_min_tensor.flat<float>()(0) = edge.input_min; in MakeInputMinMax()
514 .Attr("value", input_min_tensor) in MakeInputMinMax()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| 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()
|