Searched defs:slice_starts (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 118 const std::vector<IndexT> slice_starts, in RunR1()
149 const std::vector<IndexT> slice_starts, in RunR2()
184 const std::vector<IndexT> slice_starts, in RunR3()
377 const std::vector<IndexT> slice_starts, int expected_value_int) { in RunR0()
403 const std::vector<IndexT> slice_starts, in RunR1()
434 const std::vector<IndexT> slice_starts, in RunR2()
473 const std::vector<IndexT> slice_starts, in RunR3()
|
| D | slice_test.cc | 392 std::array<int64_t, 2> slice_starts; member
477 std::array<int64_t, 4> slice_starts; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 95 DimensionVector slice_starts(all_to_all->shape().rank(), 0); in ExpandInstruction() local
|
| D | hlo_sharding_util.cc | 716 std::vector<int64_t> slice_starts(hlo.shape().rank(), 0LL), in GatherEffectiveOutputSharding() local
846 std::vector<int64_t> slice_starts(index_rank, 0LL), slice_limits(index_rank); in ScatterEffectiveIndexSharding() local
896 std::vector<int64_t> slice_starts(data_rank, 0LL); in ScatterEffectiveDataSharding() local
|
| D | hlo_instructions.h | 1055 int64_t slice_starts(int64_t dimension) const { in slice_starts() function
1058 const std::vector<int64_t>& slice_starts() const { return slice_starts_; } in slice_starts() function
|
| D | hlo_instruction.cc | 376 std::vector<int64_t> slice_starts, slice_limits, slice_strides; in CreateFromProto() local
4498 int64_t HloInstruction::slice_starts(int64_t dimension) const { in slice_starts() function in xla::HloInstruction
4502 const std::vector<int64_t>& HloInstruction::slice_starts() const { in slice_starts() function in xla::HloInstruction
|
| D | algebraic_simplifier.cc | 4913 std::vector<int64_t> slice_starts(rank); in HandleDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | custom_call_handler.cc | 292 std::vector<int64_t> slice_starts(hlo->shape().rank(), 0); in HandleCustomCallSPMDInternal_RotateRight() 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 | 225 llvm::SmallVector<int64_t, 4> slice_starts(type.getRank(), 0); in SliceInMinorDims() local
263 auto slice_starts = CreateFullIndexVectorFromMinorIndices( in DynamicSliceInMinorDims() local
|