Home
last modified time | relevance | path

Searched defs:input_max_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_and_dequantize_op.cc86 Tensor input_max_tensor; in Compute() local
179 const Tensor& input_max_tensor = ctx->input(3); in Compute() local
274 Tensor input_max_tensor; in Compute() local
356 Tensor input_max_tensor(DataTypeToEnum<T>::value, TensorShape()); in Compute() local
H A Ddequantize_op.cc95 const Tensor& input_max_tensor = ctx->input(2); in Compute() local
H A Dquantized_batch_norm_op.cc180 const auto& input_max_tensor = context->input(2); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dquantize_training.cc514 Tensor input_max_tensor(DT_FLOAT, TensorShape()); in MakeInputMinMax() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dquantize_nodes_test.cc934 Tensor input_max_tensor(DT_FLOAT, TensorShape({})); in TestMergeAdjacentRequantizes() local