Searched refs:lhs_indices (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 242 std::vector<int> lhs_indices = {0, 0, 0, 0, 0, 0}; in TestBroadcastingExpand() local
246 lhs_indices, repeated_lhs_input, rhs_indices, repeated_rhs_input, in TestBroadcastingExpand()
271 std::vector<int> lhs_indices = {0, 1, 2, 0, 1, 2}; in TestBroadcastingInDim() local
275 lhs_indices, repeated_lhs_input, rhs_indices, repeated_rhs_input, in TestBroadcastingInDim()
300 std::vector<int> lhs_indices = {0, 0, 0, 1, 1, 1}; in TestBroadcasting() local
304 lhs_indices, repeated_lhs_input, rhs_indices, repeated_rhs_input, in TestBroadcasting()
336 std::vector<int> lhs_indices, absl::InlinedVector<T, 10> lhs_input, in ComputeExpectedOutput() argument
340 for (int i = 0; i < lhs_indices.size(); i++) { in ComputeExpectedOutput()
341 auto lhs = static_cast<BaselineT>(lhs_input[lhs_indices[i]]); in ComputeExpectedOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 76 SmallVector<Value, 2> lhs_indices{ivs[0], ivs[2]}, in matchAndRewrite() local
79 auto l = builder.create<AffineLoadOp>(loc, lhs, lhs_indices); in matchAndRewrite()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/ |
| D | cfa.h | 250 auto lhs_indices = std::make_pair(idoms[lhs.first].postorder_index, in CalculateDominators()
254 return lhs_indices < rhs_indices; in CalculateDominators()
|
| /external/deqp-deps/SPIRV-Tools/source/ |
| D | cfa.h | 250 auto lhs_indices = std::make_pair(idoms[lhs.first].postorder_index, in CalculateDominators()
254 return lhs_indices < rhs_indices; in CalculateDominators()
|