Searched defs:slice_starts (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 119 const std::vector<IndexT> slice_starts, in RunR1()
150 const std::vector<IndexT> slice_starts, in RunR2()
185 const std::vector<IndexT> slice_starts, in RunR3()
378 const std::vector<IndexT> slice_starts, int expected_value_int) { in RunR0()
404 const std::vector<IndexT> slice_starts, in RunR1()
435 const std::vector<IndexT> slice_starts, in RunR2()
474 const std::vector<IndexT> slice_starts, in RunR3()
|
| D | slice_test.cc | 390 std::array<int64, 2> slice_starts; member
475 std::array<int64, 4> slice_starts; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 94 DimensionVector slice_starts(all_to_all->shape().rank(), 0); in ExpandInstruction() local
|
| D | hlo_sharding_util.cc | 625 std::vector<int64> slice_starts(hlo.shape().rank(), 0LL), in GatherEffectiveOutputSharding() local
737 std::vector<int64> slice_starts(index_rank, 0LL), slice_limits(index_rank); in ScatterEffectiveIndexSharding() local
787 std::vector<int64> slice_starts(data_rank, 0LL); in ScatterEffectiveDataSharding() local
|
| D | hlo_instructions.h | 796 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
|
| D | hlo_instruction.cc | 311 std::vector<int64> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
3930 int64 HloInstruction::slice_starts(int64 dimension) const { in slice_starts() function in xla::HloInstruction
3934 const std::vector<int64>& HloInstruction::slice_starts() const { in slice_starts() function in xla::HloInstruction
|
| D | algebraic_simplifier.cc | 4397 std::vector<int64> slice_starts(rank); in HandleDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 273 llvm::SmallVector<int64_t, 8> slice_starts(result_shape.size(), 0); in GatherElements() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 294 llvm::SmallVector<int64_t, 4> slice_starts(type.getRank(), 0); in SliceInMinorDims() local
332 auto slice_starts = CreateFullIndexVectorFromMinorIndices( in DynamicSliceInMinorDims() local
|