Searched refs:input_rhs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 27 input_rhs, argument
29 return gen_risc_ops.risc_add(input_lhs, input_rhs, name=name)
98 def risc_div(input_lhs, input_rhs, name='RISC_DIV'): argument
99 return gen_risc_ops.risc_div(input_lhs, input_rhs, name=name)
103 input_rhs, argument
109 input_rhs,
166 def risc_max(input_lhs, input_rhs, name='RISC_MAX'): argument
167 return gen_risc_ops.risc_max(input_lhs, input_rhs, name=name)
170 def risc_min(input_lhs, input_rhs, name='RISC_MIN'): argument
171 return gen_risc_ops.risc_min(input_lhs, input_rhs, name=name)
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_matmul.cc | 79 std::string input_rhs = matmul_op->inputs[1]; in Run() local
137 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run()
142 LogName(*matmul_op), input_rhs); in Run()
146 input_rhs, in Run()
148 model, AvailableArrayName(*model, input_rhs + "/transpose/perm"), in Run()
151 AvailableArrayName(*model, input_rhs + "/transpose")}; in Run()
155 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_rhs)); in Run()
161 LogName(*matmul_op), input_rhs); in Run()
164 input_rhs = transpose_op->outputs[0]; in Run()
169 fc_op->inputs = {input_lhs, input_rhs}; in Run()
[all …]
|
| D | unroll_batch_matmul.cc | 148 std::string input_rhs = batch_op->inputs[1]; in Run() local
150 const auto& input_rhs_array = model->GetArray(input_rhs); in Run()
164 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run()
166 input_rhs = transpose_op->outputs[0]; in Run()
168 const auto& input_array_b = model->GetArray(input_rhs); in Run()
192 matmul_op->inputs = {input_lhs, input_rhs}; in Run()
208 SliceInput(input_rhs, base_name, "b", bcast.y_batch_size(), input_array_b, in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 54 Value input_rhs = op.y(); in matchAndRewrite() local
58 auto rhs_type = input_rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
79 /*inputs=*/ValueRange({input_lhs, input_rhs}), in matchAndRewrite()
|
| D | unroll_batch_matmul.cc | 148 Value input_rhs = op.y(); in matchAndRewrite() local
154 if (!input_rhs.getType().isa<RankedTensorType>()) { in matchAndRewrite()
160 auto rhs_type = input_rhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
212 /*b=*/input_rhs, in matchAndRewrite()
244 sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); in matchAndRewrite()
|