Searched defs:input_rhs (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/risc/ |
H A D | risc_ops.py | 27 input_rhs, argument 98 def risc_div(input_lhs, input_rhs, name='RISC_DIV'): argument 103 input_rhs, argument 166 def risc_max(input_lhs, input_rhs, name='RISC_MAX'): argument 170 def risc_min(input_lhs, input_rhs, name='RISC_MIN'): argument 174 def risc_mul(input_lhs, input_rhs, name='RISC_MUL'): argument 191 def risc_pow(input_lhs, input_rhs, name='RISC_POW'): argument 235 def risc_sub(input_lhs, input_rhs, name='RISC_SUB'): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | batchmatmul_to_einsum.cc | 54 Value input_rhs = op.y(); in matchAndRewrite() local
|
H A D | unroll_batch_matmul.cc | 148 Value input_rhs = op.y(); in matchAndRewrite() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
H A D | resolve_tensorflow_matmul.cc | 79 std::string input_rhs = matmul_op->inputs[1]; in Run() local
|
H A D | unroll_batch_matmul.cc | 148 std::string input_rhs = batch_op->inputs[1]; in Run() local
|