Searched refs:input_lhs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 26 input_lhs, 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)
102 def risc_dot(input_lhs, argument
108 input_lhs,
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 | 78 std::string input_lhs = matmul_op->inputs[0]; in Run() local
84 Array& lhs_array = model->GetArray(input_lhs); in Run()
111 input_lhs, in Run()
113 model, AvailableArrayName(*model, input_lhs + "/transpose/perm"), in Run()
116 AvailableArrayName(*model, input_lhs + "/transpose")}; in Run()
120 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_lhs)); in Run()
121 input_lhs = transpose_op->outputs[0]; in Run()
169 fc_op->inputs = {input_lhs, input_rhs}; in Run()
213 input_lhs = previous_op->inputs[0]; in Run()
214 fc_op->inputs = {input_lhs, input_rhs}; in Run()
|
| D | unroll_batch_matmul.cc | 147 std::string input_lhs = batch_op->inputs[0]; in Run() local
149 const auto& input_lhs_array = model->GetArray(input_lhs); in Run()
156 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run()
158 input_lhs = transpose_op->outputs[0]; in Run()
160 const auto& input_array_a = model->GetArray(input_lhs); in Run()
192 matmul_op->inputs = {input_lhs, input_rhs}; in Run()
205 SliceInput(input_lhs, base_name, "a", bcast.x_batch_size(), input_array_a, in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 53 Value input_lhs = op.x(); in matchAndRewrite() local
57 auto lhs_type = input_lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
79 /*inputs=*/ValueRange({input_lhs, input_rhs}), in matchAndRewrite()
|
| D | unroll_batch_matmul.cc | 147 Value input_lhs = op.x(); in matchAndRewrite() local
150 if (!input_lhs.getType().isa<RankedTensorType>()) { in matchAndRewrite()
159 auto lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
211 /*a=*/input_lhs, in matchAndRewrite()
242 sliceInput(input_lhs, bcast.x_batch_size(), loc, rewriter); in matchAndRewrite()
|