Searched refs:rhs_input (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 46 const absl::InlinedVector<T, 10>& rhs_input, in SetOpKernel() argument
61 AddInputFromArray<T>(rhs_shape, rhs_input); in SetOpKernel()
71 const absl::InlinedVector<T, 10>& rhs_input, in RunAndExpectResult() argument
75 SetOpKernel<T, OutT>(op_name, lhs_shape, lhs_input, rhs_shape, rhs_input, in RunAndExpectResult()
96 const absl::InlinedVector<T, 10>& rhs_input, in RunAndExpectInvalidArgument() argument
98 SetOpKernel<T, OutT>(op_name, lhs_shape, lhs_input, rhs_shape, rhs_input, in RunAndExpectInvalidArgument()
110 const absl::InlinedVector<T, 10>& rhs_input, in TestIncompatibleShapes() argument
118 test::RepeatInputToMatchShape(rhs_input, rhs_shape.num_elements()); in TestIncompatibleShapes()
128 const absl::InlinedVector<T, 10>& rhs_input, in TestEqualShapes() argument
133 CHECK(lhs_input.size() <= input_size && rhs_input.size() <= input_size && in TestEqualShapes()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 869 DenseFPElementsAttr rhs_input; in matchAndRewrite() local
871 !matchPattern(rw_rhs.operand(), m_Constant(&rhs_input)) || in matchAndRewrite()
872 !rhs_input.isSplat() || in matchAndRewrite()
873 !rhs_input.getSplatValue<APFloat>().isExactlyValue(1.0)) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.cc | 589 const int64 out_buffer_size, const T* lhs_block, const T* rhs_input, in GemmState() argument
598 rhs_mapper_(rhs_input, depth_), in GemmState()
|