Searched refs:max_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 | 36 const Tensor& max_input_tensor = context->input(2); in Compute() local
43 context, TensorShapeUtils::IsScalar(max_input_tensor.shape()), in Compute()
45 max_input_tensor.dims())); in Compute()
48 const float max_input = max_input_tensor.scalar<float>()(); in Compute()
82 const Tensor& max_input_tensor = context->input(2); in Compute() local
89 context, TensorShapeUtils::IsScalar(max_input_tensor.shape()), in Compute()
91 max_input_tensor.dims())); in Compute()
94 const float max_input = max_input_tensor.scalar<float>()(); in Compute()
|
| H A D | quantized_pooling_ops.cc | 71 const Tensor& max_input_tensor = context->input(2); in Compute() local
77 OP_REQUIRES(context, TensorShapeUtils::IsScalar(max_input_tensor.shape()), in Compute()
80 max_input_tensor.dims(), in Compute()
81 ", received shape: ", max_input_tensor.shape())); in Compute()
135 const Tensor& max_input_tensor = context->input(2); in Compute() local
141 OP_REQUIRES(context, TensorShapeUtils::IsScalar(max_input_tensor.shape()), in Compute()
144 max_input_tensor.dims(), in Compute()
145 ", received shape: ", max_input_tensor.shape())); in Compute()
|