| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_lib_cpu.cc | 32 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes) { in operator ()()
|
| D | split_lib_gpu.cu.cc | 37 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes) { in operator ()()
46 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_sizes) { in operator ()()
|
| D | split_op.cc | 166 Eigen::DSizes<Eigen::DenseIndex, NDims> slice_sizes; in operator ()() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | get_item_op.cc | 55 std::vector<int64> slice_sizes = {1}; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 117 auto slice_sizes = input_shape.dim_sizes(); in Compile() local
|
| D | gather_scatter_ops.cc | 38 std::vector<int64> slice_sizes; in Compile() local
|
| D | fft_ops.cc | 61 absl::InlinedVector<int64, 4> slice_sizes = input_shape.dim_sizes(); in Compile() local
|
| D | gather_op.cc | 123 std::vector<int64> slice_sizes; in XlaGather() local
|
| D | strided_slice_op.cc | 104 std::vector<int64> slice_sizes(input_shape.dims(), 1); in EmitDynamicSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 215 std::vector<int64> slice_sizes(input_shape.rank(), 1); in TorchGather() local
278 std::vector<int64> slice_sizes = SpanToVector(input_shape.dimensions()); in TorchIndexSelect() local
|
| D | matrix.cc | 150 std::vector<int64> slice_sizes; in GetMatrixDiagonalViaGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 234 absl::Span<const int64> slice_sizes, int64 gather_loop_trip_count, in CreateGatherLoopAccumulatorInitValue()
|
| D | hlo_creation_utils.cc | 137 absl::Span<const int64> slice_sizes) { in MakeDynamicSliceHlo()
152 absl::Span<const int64> slice_sizes) { in MakeDynamicSliceHlo()
|
| D | triangular_solve_expander.cc | 76 std::vector<int64> slice_sizes(ndims); in DiagonalBlocks() local
|
| D | hlo_instruction.cc | 626 std::vector<int64> slice_sizes(proto.dynamic_slice_sizes_size()); in CreateFromProto() local
1241 absl::Span<const int64> slice_sizes) { in CreateDynamicSlice()
1618 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in CreateGather()
4228 int64 HloInstruction::slice_sizes(int64 dimension) const { in slice_sizes() function in xla::HloInstruction
|
| D | shape_inference.cc | 2475 absl::Span<const int64> slice_sizes, bool allow_scalar_indices) { in InferDynamicSliceShape()
3305 absl::Span<const int64> slice_sizes) { in InferGatherShape()
|
| D | indexed_array_analysis.cc | 255 absl::Span<const int64> slice_sizes, Array* source, Array* indices) { in ComputeArrayForGather()
|
| D | hlo_instructions.h | 1611 int64 slice_sizes(int64 dimension) const { in slice_sizes() function
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 353 std::vector<int64_t> slice_sizes( in ImportInstructionImpl() local
365 std::vector<int64_t> slice_sizes( in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 120 const std::vector<int64>& slice_sizes, in RunR1()
151 const std::vector<int64>& slice_sizes, in RunR2()
186 const std::vector<int64>& slice_sizes, in RunR3()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 290 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal()
519 absl::Span<const int64> slice_sizes) { in DynamicSliceInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 303 auto slice_sizes = GetR1Const(per_slice_shape, builder, loc); in ScatterAccumulateElements() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 482 def gather(operand, start_indices, dimension_numbers, slice_sizes, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 335 auto slice_sizes = llvm::to_vector<4>(type.getShape()); in DynamicSliceInMinorDims() local
985 DenseIntElementsAttr slice_sizes) { in CanBeTranslatedToDynamicSlice()
1012 Value input, Value start_indices, DenseIntElementsAttr slice_sizes, in TFSliceSizes2HLOSliceSizes()
1634 Shape slice_sizes(input_shape.begin(), input_shape.end()); in matchAndRewrite() local
3251 SmallVector<int64_t, 4> slice_sizes(slicing_dim_size, 1); in rewriteWithUnknownBegin() local
5126 SmallVector<int64_t, 4> slice_sizes(input_shape.begin(), input_shape.end()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 996 absl::Span<const int64> slice_sizes) { in DynamicSlice()
1014 absl::Span<const int64> slice_sizes) { in DynamicSliceInternal()
2272 absl::Span<const int64> slice_sizes, in Gather()
2289 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal()
4168 absl::Span<const int64> slice_sizes) { in DynamicSlice()
4877 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in Gather()
|