Searched refs:rhs_indices (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 243 std::vector<int> rhs_indices = {0, 1, 2, 3, 4, 5}; in TestBroadcastingExpand() local
246 lhs_indices, repeated_lhs_input, rhs_indices, repeated_rhs_input, in TestBroadcastingExpand()
272 std::vector<int> rhs_indices = {0, 1, 2, 3, 4, 5}; in TestBroadcastingInDim() local
275 lhs_indices, repeated_lhs_input, rhs_indices, repeated_rhs_input, in TestBroadcastingInDim()
301 std::vector<int> rhs_indices = {0, 1, 2, 0, 1, 2}; in TestBroadcasting() local
304 lhs_indices, repeated_lhs_input, rhs_indices, repeated_rhs_input, in TestBroadcasting()
337 std::vector<int> rhs_indices, absl::InlinedVector<T, 10> rhs_input, in ComputeExpectedOutput() argument
342 auto rhs = static_cast<BaselineT>(rhs_input[rhs_indices[i]]); in ComputeExpectedOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 77 rhs_indices{ivs[2], ivs[1]}, result_indices{ivs[0], ivs[1]}; in matchAndRewrite() local
80 auto r = builder.create<AffineLoadOp>(loc, rhs, rhs_indices); in matchAndRewrite()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/ |
| D | cfa.h | 252 auto rhs_indices = std::make_pair(idoms[rhs.first].postorder_index, in CalculateDominators()
254 return lhs_indices < rhs_indices; in CalculateDominators()
|
| /external/deqp-deps/SPIRV-Tools/source/ |
| D | cfa.h | 252 auto rhs_indices = std::make_pair(idoms[rhs.first].postorder_index, in CalculateDominators()
254 return lhs_indices < rhs_indices; in CalculateDominators()
|