Searched refs:input_lhs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 30 input_lhs, 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)
106 def risc_dot(input_lhs, argument
112 input_lhs,
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/ |
| D | resolve_tensorflow_matmul.cc | 77 std::string input_lhs = matmul_op->inputs[0]; in Run() local
83 Array& lhs_array = model->GetArray(input_lhs); in Run()
110 input_lhs, in Run()
112 model, AvailableArrayName(*model, input_lhs + "/transpose/perm"), in Run()
115 AvailableArrayName(*model, input_lhs + "/transpose")}; in Run()
119 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_lhs)); in Run()
120 input_lhs = transpose_op->outputs[0]; in Run()
168 fc_op->inputs = {input_lhs, input_rhs}; in Run()
212 input_lhs = previous_op->inputs[0]; in Run()
213 fc_op->inputs = {input_lhs, input_rhs}; in Run()
|
| D | unroll_batch_matmul.cc | 145 std::string input_lhs = batch_op->inputs[0]; in Run() local
147 const auto& input_lhs_array = model->GetArray(input_lhs); in Run()
154 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run()
156 input_lhs = transpose_op->outputs[0]; in Run()
158 const auto& input_array_a = model->GetArray(input_lhs); in Run()
190 matmul_op->inputs = {input_lhs, input_rhs}; in Run()
203 SliceInput(input_lhs, base_name, "a", bcast.x_batch_size(), input_array_a, in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 192 Value input_lhs = op.x(); in matchAndRewrite() local
195 if (!input_lhs.getType().isa<RankedTensorType>()) { in matchAndRewrite()
204 auto lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
229 input_lhs = createTransposeOp(input_lhs, loc, rewriter); in matchAndRewrite()
231 lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
254 /*a=*/input_lhs, in matchAndRewrite()
286 sliceInput(input_lhs, bcast.x_batch_size(), loc, rewriter); in matchAndRewrite()
|
| D | batchmatmul_to_einsum.cc | 52 Value input_lhs = op.x(); in matchAndRewrite() local
56 auto lhs_type = input_lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
78 /*inputs=*/ValueRange({input_lhs, input_rhs}), in matchAndRewrite()
|