Searched refs:slice0 (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | slice_sinker_test.cc | 69 const HloInstruction* slice0; in TEST_F() local
73 m::Slice(&slice0, m::Select(m::Parameter(0), m::Parameter(1), in TEST_F()
77 EXPECT_THAT(slice0->slice_starts(), ElementsAre(0, 0)); in TEST_F()
78 EXPECT_THAT(slice0->slice_limits(), ElementsAre(2, 9)); in TEST_F()
79 EXPECT_THAT(slice0->slice_strides(), ElementsAre(1, 1)); in TEST_F()
109 const HloInstruction* slice0; in TEST_F() local
114 m::Slice(&slice0, m::Add(m::Parameter(0), m::Parameter(1))), in TEST_F()
117 EXPECT_THAT(slice0->slice_starts(), ElementsAre(0, 0)); in TEST_F()
118 EXPECT_THAT(slice0->slice_limits(), ElementsAre(2, 9)); in TEST_F()
119 EXPECT_THAT(slice0->slice_strides(), ElementsAre(1, 1)); in TEST_F()
[all …]
|
| D | slice_sinker.cc | 58 const HloInstruction* slice0 = inst->operand(0); in IsElementwiseOperationOnSimilarSlices() local
60 [slice0](const HloInstruction* slice) { in IsElementwiseOperationOnSimilarSlices()
62 slice0->operand(0)->shape(), in IsElementwiseOperationOnSimilarSlices()
64 SameSliceConfiguration(slice0, slice); in IsElementwiseOperationOnSimilarSlices()
|
| D | buffer_assignment_test.cc | 2367 TF_ASSERT_OK_AND_ASSIGN(auto slice0, assignment->GetUniqueSlice(tuple, {0})); in TEST_F()
2369 EXPECT_NE(slice0, slice1); in TEST_F()
2382 EXPECT_EQ(slice0, slice_while2); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 2345 HloInstruction* slice0 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F() local
2392 {slice0, slice1, slice2, slice3, slice4, slice5, slice6, slice7, slice8}, in TEST_F()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SaveSlices.pbtxt | 44 `slice-spec` itself is a `:`-separated list: `slice0:slice1:...:sliceN-1`
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | fused_slice.hlo | 63 // CHECK: slice0-after: ; preds = %[[VAL_60]], %[[VAL_45]]
75 // CHECK: slice0-true: ; preds = %[[VAL_45]]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamism_inference_test.cc | 189 auto slice0 = SliceInDim(concat, 0, 1, 1, 0); in TEST_F() local
190 auto reshape0 = Reshape(slice0, {}); in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 501 slice0 = _IotaInitializer([5, 5])
505 self.assertAllClose(slice0 + slice1 + slice2, val)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 12817 `slice-spec` itself is a `:`-separated list: `slice0:slice1:...:sliceN-1`
|