Home
last modified time | relevance | path

Searched refs:input_rhs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/risc/
Drisc_ops.py31 input_rhs, argument
33 return gen_risc_ops.risc_add(input_lhs, input_rhs, name=name)
102 def risc_div(input_lhs, input_rhs, name='RISC_DIV'): argument
103 return gen_risc_ops.risc_div(input_lhs, input_rhs, name=name)
107 input_rhs, argument
113 input_rhs,
170 def risc_max(input_lhs, input_rhs, name='RISC_MAX'): argument
171 return gen_risc_ops.risc_max(input_lhs, input_rhs, name=name)
174 def risc_min(input_lhs, input_rhs, name='RISC_MIN'): argument
175 return gen_risc_ops.risc_min(input_lhs, input_rhs, name=name)
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_tensorflow_matmul.cc78 std::string input_rhs = matmul_op->inputs[1]; in Run() local
136 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run()
141 LogName(*matmul_op), input_rhs); in Run()
145 input_rhs, in Run()
147 model, AvailableArrayName(*model, input_rhs + "/transpose/perm"), in Run()
150 AvailableArrayName(*model, input_rhs + "/transpose")}; in Run()
154 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_rhs)); in Run()
160 LogName(*matmul_op), input_rhs); in Run()
163 input_rhs = transpose_op->outputs[0]; in Run()
168 fc_op->inputs = {input_lhs, input_rhs}; in Run()
[all …]
Dunroll_batch_matmul.cc146 std::string input_rhs = batch_op->inputs[1]; in Run() local
148 const auto& input_rhs_array = model->GetArray(input_rhs); in Run()
162 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run()
164 input_rhs = transpose_op->outputs[0]; in Run()
166 const auto& input_array_b = model->GetArray(input_rhs); in Run()
190 matmul_op->inputs = {input_lhs, input_rhs}; in Run()
206 SliceInput(input_rhs, base_name, "b", bcast.y_batch_size(), input_array_b, in Run()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc193 Value input_rhs = op.y(); in matchAndRewrite() local
199 if (!input_rhs.getType().isa<RankedTensorType>()) { in matchAndRewrite()
205 auto rhs_type = input_rhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
237 input_rhs = createTransposeOp(input_rhs, loc, rewriter); in matchAndRewrite()
239 rhs_type = input_rhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
255 /*b=*/input_rhs, in matchAndRewrite()
288 sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); in matchAndRewrite()
Dbatchmatmul_to_einsum.cc53 Value input_rhs = op.y(); in matchAndRewrite() local
57 auto rhs_type = input_rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
78 /*inputs=*/ValueRange({input_lhs, input_rhs}), in matchAndRewrite()