Searched refs:min_input_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_activation_ops.cc | 35 const Tensor& min_input_tensor = context->input(1); in Compute() local
39 context, TensorShapeUtils::IsScalar(min_input_tensor.shape()), in Compute()
41 min_input_tensor.dims())); in Compute()
47 const float min_input = min_input_tensor.scalar<float>()(); in Compute()
81 const Tensor& min_input_tensor = context->input(1); in Compute() local
85 context, TensorShapeUtils::IsScalar(min_input_tensor.shape()), in Compute()
87 min_input_tensor.dims())); in Compute()
93 const float min_input = min_input_tensor.scalar<float>()(); in Compute()
|
| H A D | quantized_pooling_ops.cc | 70 const Tensor& min_input_tensor = context->input(1); in Compute() local
72 OP_REQUIRES(context, TensorShapeUtils::IsScalar(min_input_tensor.shape()), in Compute()
75 min_input_tensor.dims(), in Compute()
76 ", received shape: ", min_input_tensor.shape())); in Compute()
134 const Tensor& min_input_tensor = context->input(1); in Compute() local
136 OP_REQUIRES(context, TensorShapeUtils::IsScalar(min_input_tensor.shape()), in Compute()
139 min_input_tensor.dims(), in Compute()
140 ", received shape: ", min_input_tensor.shape())); in Compute()
|