Searched defs:scalar_input (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_add_op.cc | 48 int64_t num_elements, T scalar_input, in ScalarAddition()
65 int64 num_elements, quint8 scalar_input, in ScalarAddition()
122 int64_t num_elements, quint8 scalar_input, in ScalarAddition()
|
| H A D | quantized_mul_op.cc | 38 T scalar_input, int32_t scalar_input_offset, in ScalarMultiply()
54 quint8 scalar_input, in ScalarMultiply()
|
| H A D | example_parsing_ops_test.cc | 206 bool scalar_input = (batch_size == 0); in ParseExampleV2() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| H A D | base_binary_ops_test.h | 162 void TestOneScalar(const std::string& op_name, T scalar_input, in TestOneScalar()
194 void TestOneEffectiveScalar(const std::string& op_name, T scalar_input, in TestOneEffectiveScalar()
|