Home
last modified time | relevance | path

Searched refs:slice_starts (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dslice_sinker_test.cc77 EXPECT_THAT(slice0->slice_starts(), ElementsAre(0, 0)); in TEST_F()
80 EXPECT_THAT(slice1->slice_starts(), ElementsAre(2, 0)); in TEST_F()
117 EXPECT_THAT(slice0->slice_starts(), ElementsAre(0, 0)); in TEST_F()
120 EXPECT_THAT(slice1->slice_starts(), ElementsAre(3, 0)); in TEST_F()
123 EXPECT_THAT(slice2->slice_starts(), ElementsAre(0, 0)); in TEST_F()
166 EXPECT_THAT(slice0->slice_starts(), ElementsAre(0, 0)); in TEST_F()
169 EXPECT_THAT(slice1->slice_starts(), ElementsAre(2, 0)); in TEST_F()
172 EXPECT_THAT(slice2->slice_starts(), ElementsAre(0, 0)); in TEST_F()
175 EXPECT_THAT(slice3->slice_starts(), ElementsAre(0, 2)); in TEST_F()
212 EXPECT_THAT(slice0->slice_starts(), ElementsAre(0, 0)); in TEST_F()
[all …]
Dall_to_all_decomposer.cc94 DimensionVector slice_starts(all_to_all->shape().rank(), 0); in ExpandInstruction() local
107 slice_shape, operand, slice_starts, slice_limits, slice_strides))); in ExpandInstruction()
109 slice_starts[split_dim] = slice_limits[split_dim]; in ExpandInstruction()
Ddynamic_index_splitter_test.cc64 EXPECT_EQ(slice->slice_starts(0), i); in TEST_F()
102 EXPECT_EQ(slice->slice_starts(0), i); in TEST_F()
Dslice_sinker.cc39 return slice_1->slice_starts() == slice_2->slice_starts() && in SameSliceConfiguration()
Dalgebraic_simplifier.cc1258 operands[j]->slice_starts(concatenate_dimension) == slice_end) { in HandleConcatenate()
1264 for (int64 k = 0; k < operands[j]->slice_starts().size(); ++k) { in HandleConcatenate()
1268 if (operands[i]->slice_starts(k) != operands[j]->slice_starts(k)) { in HandleConcatenate()
1283 slice_end - operands[i]->slice_starts(concatenate_dimension)); in HandleConcatenate()
1290 /*start_indices=*/operands[i]->slice_starts(), in HandleConcatenate()
4012 if (next_operand_start > slice->slice_starts(0)) { in TrySimplifyScalarSlice()
4028 {slice->slice_starts(0) - operand_start}, in TrySimplifyScalarSlice()
4029 {slice->slice_starts(0) - operand_start + 1}, in TrySimplifyScalarSlice()
4052 if (slice->slice_starts(i) != 0 || in TryToReorderSliceAndReshape()
4059 slice->slice_starts(0) == 0) { in TryToReorderSliceAndReshape()
[all …]
Dhlo_cse.cc124 tensorflow::Hash64Combine(hash, c_hash(instruction->slice_starts())), in CseHash()
Dhlo_sharding_util.cc625 std::vector<int64> slice_starts(hlo.shape().rank(), 0LL), in GatherEffectiveOutputSharding() local
635 hlo.sharding().tile_assignment().Slice(slice_starts, slice_limits); in GatherEffectiveOutputSharding()
737 std::vector<int64> slice_starts(index_rank, 0LL), slice_limits(index_rank); in ScatterEffectiveIndexSharding() local
746 index_sharding.tile_assignment().Slice(slice_starts, slice_limits); in ScatterEffectiveIndexSharding()
787 std::vector<int64> slice_starts(data_rank, 0LL); in ScatterEffectiveDataSharding() local
789 data_sharding.tile_assignment().Slice(slice_starts, tile_assignment_dims); in ScatterEffectiveDataSharding()
Dtopk_rewriter.cc109 if (absl::c_any_of(slice->slice_starts(), in Run()
Dhlo_instruction.cc311 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
314 slice_starts.push_back(slice_dimensions.start()); in CreateFromProto()
318 instruction = CreateSlice(shape, operands(0), slice_starts, slice_limits, in CreateFromProto()
3930 int64 HloInstruction::slice_starts(int64 dimension) const { in slice_starts() function in xla::HloInstruction
3931 return Cast<HloSliceInstruction>(this)->slice_starts(dimension); in slice_starts()
3934 const std::vector<int64>& HloInstruction::slice_starts() const { in slice_starts() function in xla::HloInstruction
3935 return Cast<HloSliceInstruction>(this)->slice_starts(); in slice_starts()
Dhlo_instruction.h1687 int64 slice_starts(int64 dimension) const;
1688 const std::vector<int64>& slice_starts() const;
Dalgebraic_simplifier_test.cc2152 EXPECT_EQ(slice->slice_starts(0), 5); in TEST_F()
2154 EXPECT_EQ(slice->slice_starts(1), 0); in TEST_F()
2156 EXPECT_EQ(slice->slice_starts(2), 0); in TEST_F()
2188 EXPECT_EQ(slice->slice_starts(0), 5); in TEST_F()
2190 EXPECT_EQ(slice->slice_starts(1), 2); in TEST_F()
2192 EXPECT_EQ(slice->slice_starts(2), 1); in TEST_F()
2407 EXPECT_EQ(computation->root_instruction()->operand(3)->slice_starts(1), 40); in TEST_F()
3475 EXPECT_EQ(computation->root_instruction()->slice_starts(0), 3); in TEST_F()
3476 EXPECT_EQ(computation->root_instruction()->slice_starts(1), 5); in TEST_F()
4860 EXPECT_THAT(root->slice_starts(), ElementsAre(1, 1)); in TEST_F()
[all …]
Dhlo_instructions.h796 int64 slice_starts(int64 dimension) const { return slice_starts_[dimension]; } in slice_starts() function
797 const std::vector<int64>& slice_starts() const { return slice_starts_; } in slice_starts() function
Ddynamic_dimension_inference.cc1102 if (hlo->slice_starts(dimension) != 0 || in HandleSlice()
Dhlo_evaluator_typed_visitor.h2440 operand->shape(), slice->slice_starts(),
2453 slice->slice_starts(i) + out_index[i] * slice->slice_strides(i);
Dhlo_verifier.cc841 slice->operand(0)->shape(), slice->slice_starts(), in HandleSlice()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc119 const std::vector<IndexT> slice_starts, in RunR1() argument
140 slice_starts[0], 0, "slice_starts", &builder, &starts); in RunR1()
150 const std::vector<IndexT> slice_starts, in RunR2() argument
168 slice_starts[i], i, "slice_starts", &builder, &starts[i]); in RunR2()
185 const std::vector<IndexT> slice_starts, in RunR3() argument
203 slice_starts[i], i, "slice_starts", &builder, &starts[i]); in RunR3()
378 const std::vector<IndexT> slice_starts, int expected_value_int) { in RunR0() argument
404 const std::vector<IndexT> slice_starts, in RunR1() argument
423 slice_starts[0], 0, "slice_starts", &builder, &starts); in RunR1()
435 const std::vector<IndexT> slice_starts, in RunR2() argument
[all …]
Dslice_test.cc390 std::array<int64, 2> slice_starts; member
410 Slice(a, spec.slice_starts, spec.slice_limits, spec.slice_strides); in XLA_TEST_P()
415 input, spec.slice_starts, spec.slice_limits, spec.slice_strides); in XLA_TEST_P()
475 std::array<int64, 4> slice_starts; member
484 "__starts_", absl::StrJoin(spec.slice_starts, "x"), in R4SpecToString()
497 values, spec.slice_starts, spec.slice_limits, spec.slice_strides); in Run()
504 Slice(parameter, spec.slice_starts, spec.slice_limits, spec.slice_strides); in Run()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.cc273 llvm::SmallVector<int64_t, 8> slice_starts(result_shape.size(), 0); in GatherElements() local
274 slice_starts[0] = maybe_contiguous_start; in GatherElements()
279 ArrayRef<Value>{buffer, GetR1Const(slice_starts, builder, loc), in GatherElements()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc481 ConvertDimensions(instruction->slice_starts()), in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc294 llvm::SmallVector<int64_t, 4> slice_starts(type.getRank(), 0); in SliceInMinorDims() local
297 slice_starts.begin() + major_dims); in SliceInMinorDims()
303 GetI64ElementsAttr(slice_starts, builder), in SliceInMinorDims()
332 auto slice_starts = CreateFullIndexVectorFromMinorIndices( in DynamicSliceInMinorDims() local
340 loc, slice_type, v, slice_starts, in DynamicSliceInMinorDims()
Dmhlo_to_lhlo_with_xla.cc1234 slice.start_indicesAttr(builder_.getI64TensorAttr(hlo_slice->slice_starts())); in EmitSliceOp()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1903 ? slice->slice_limits(memcpy_dim) - slice->slice_starts(memcpy_dim) in HandleSlice()
1940 /*operand_shape=*/operand->shape(), /*starts=*/slice->slice_starts(), in HandleSlice()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5760 for (size_t dim = 0; dim < slice->slice_starts().size(); ++dim) { in EmitElementForInputFusibleSlices()
5764 index.GetConstantWithIndexType(slice->slice_starts(dim))); in EmitElementForInputFusibleSlices()
5780 index.GetConstantWithIndexType(slice->slice_starts(dim))); in EmitElementForInputFusibleSlices()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1595 dim->set_padding_low(-hlo->slice_starts(i)); in HandleSlice()
1611 std::vector<int64> start_indices = hlo->slice_starts(); in HandleSlice()
Dspmd_partitioner_util.cc1194 if (absl::c_any_of(slice->slice_starts(), [](int x) { return x != 0; }) || in GetKValueInTopKWhenPartitionSortDim()

12