Searched refs:dims0 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 162 auto* dims0 = padding_config.add_dimensions(); in TEST_F() local
163 dims0->set_edge_padding_low(1); in TEST_F()
164 dims0->set_interior_padding(0); in TEST_F()
165 dims0->set_edge_padding_high(2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | chlo_infer_shape_type_methods.mlir | 22 // CHECK: "mhlo_test.return_type_components"(%0) {dims0 = [-1, -1], element_type0 = complex<f32>}
31 // CHECK: "mhlo_test.return_type_components"(%0) {dims0 = [-1, -1], element_type0 = i1}
40 // CHECK: "mhlo_test.return_type_components"(%0) {dims0 = [-1], element_type0 = f32}
51 // CHECK: "mhlo_test.return_type_components"(%0) {dims0 = [-1, -1], element_type0 = f32}
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 605 std::vector<int64> dims0({0}); in TEST_F() local
608 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation)); in TEST_F()
|