HomeSort by relevance Sort by last modified time
    Searched refs:input_max_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,
38 input_max_tensor, out);
quantize_and_dequantize_op.cc 61 Tensor input_max_tensor; variable
64 input_max_tensor = ctx->input(2);
66 auto max_val = input_max_tensor.scalar<T>()();
74 TensorShape(), &input_max_tensor));
79 range_given_, &input_min_tensor, &input_max_tensor, output->flat<T>());
120 Tensor input_max_tensor; variable
123 input_max_tensor = ctx->input(2);
125 auto max_val = input_max_tensor.scalar<T>()();
133 TensorShape(), &input_max_tensor));
138 range_given_, &input_min_tensor, &input_max_tensor, output->flat<T>())
    [all...]
quantize_and_dequantize_op.h 30 Tensor* input_min_tensor, Tensor* input_max_tensor,
39 Tensor* input_min_tensor, Tensor* input_max_tensor,
44 auto input_max = input_max_tensor->scalar<T>();
  /external/tensorflow/tensorflow/core/graph/
quantize_training.cc 510 Tensor input_max_tensor(DT_FLOAT, TensorShape());
511 input_max_tensor.flat<float>()(0) = edge.input_max;
515 .Attr("value", input_max_tensor)
  /external/tensorflow/tensorflow/tools/graph_transforms/
quantize_nodes_test.cc     [all...]

Completed in 348 milliseconds