HomeSort by relevance Sort by last modified time
    Searched refs:input_min_tensor (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
quantize_and_dequantize_op_gpu.cu.cc 34 Tensor* input_min_tensor, Tensor* input_max_tensor,
37 d, input, signed_input, num_bits, range_given, input_min_tensor,
quantize_and_dequantize_op.cc 75 Tensor input_min_tensor; variable
78 input_min_tensor = ctx->input(1);
80 auto min_val = input_min_tensor.scalar<T>()();
87 TensorShape(), &input_min_tensor));
94 range_given_, &input_min_tensor, &input_max_tensor, round_mode_,
136 Tensor input_min_tensor; variable
139 input_min_tensor = ctx->input(1);
141 auto min_val = input_min_tensor.scalar<T>()();
148 TensorShape(), &input_min_tensor));
155 range_given_, &input_min_tensor, &input_max_tensor, ROUND_HALF_TO_EVEN
    [all...]
quantize_and_dequantize_op.h 47 Tensor* input_min_tensor, Tensor* input_max_tensor,
109 Tensor* input_min_tensor, Tensor* input_max_tensor,
114 auto input_min = input_min_tensor->scalar<T>();
123 min_range = input_min_tensor->scalar<T>()();
  /external/tensorflow/tensorflow/core/graph/
quantize_training.cc 509 Tensor input_min_tensor(DT_FLOAT, TensorShape());
510 input_min_tensor.flat<float>()(0) = edge.input_min;
514 .Attr("value", input_min_tensor)
  /external/tensorflow/tensorflow/tools/graph_transforms/
quantize_nodes_test.cc     [all...]

Completed in 1074 milliseconds