| /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 | 38 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes) { in operator ()()
47 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_sizes) { in operator ()()
|
| D | split_op.cc | 169 Eigen::DSizes<Eigen::DenseIndex, NDims> slice_sizes; in operator ()() local
379 Eigen::DSizes<Eigen::DenseIndex, 3> slice_sizes; in Compute() local
|
| D | batch_kernels.cc | 162 Eigen::DSizes<Eigen::DenseIndex, 2> slice_sizes{size, suffix_dim_size}; in SplitCPU() local
360 const std::vector<int64> slice_sizes = {1}; in ConcatInputTensors() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 117 auto slice_sizes = input_shape.dim_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 | 117 std::vector<int64> slice_sizes; in XlaGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 153 std::vector<int64> slice_sizes(input_shape.rank(), 1); in TorchGather() local
169 std::vector<int64> slice_sizes = input_shape.dimensions(); in TorchIndexSelect() 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 | triangular_solve_expander.cc | 73 std::vector<int64> slice_sizes(ndims); in DiagonalBlocks() local
|
| D | hlo_creation_utils.cc | 123 absl::Span<const int64> slice_sizes) { in MakeDynamicSliceHlo()
|
| D | hlo_instruction.cc | 483 std::vector<int64> slice_sizes(proto.dynamic_slice_sizes_size()); in CreateFromProto() local
995 absl::Span<const int64> slice_sizes) { in CreateDynamicSlice()
1309 absl::Span<const int64> slice_sizes) { in CreateGather()
3508 int64 HloInstruction::slice_sizes(int64 dimension) const { in slice_sizes() function in xla::HloInstruction
|
| D | hlo_instructions.h | 1318 int64 slice_sizes(int64 dimension) const { in slice_sizes() function
|
| D | shape_inference.cc | 2267 absl::Span<const int64> slice_sizes, bool allow_scalar_indices) { in InferDynamicSliceShape()
2960 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.cc | 2235 absl::Span<const int64> slice_sizes) in HloDynamicSliceInstruction()
2245 absl::Span<const int64> slice_sizes) in HloDynamicSliceInstruction()
2312 absl::Span<const int64> slice_sizes) in HloGatherInstruction()
|
| D | hlo_evaluator.cc | 800 int64 output_rank, absl::Span<const int64> slice_sizes, in IterationSpaceForOutputOffsetIndices()
|
| D | hlo_parser.cc | 1627 optional<std::vector<int64>> slice_sizes; in ParseInstructionRhs() local
|
| D | algebraic_simplifier_test.cc | 4657 std::vector<int64> slice_sizes = {slice_row_size, slice_col_size}; in TEST_P() local
4738 std::vector<int64> slice_sizes = {slice_row_size, slice_col_size}; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamic_ops_test.cc | 119 const std::vector<int64>& slice_sizes, in RunR1()
150 const std::vector<int64>& slice_sizes, in RunR2()
185 const std::vector<int64>& slice_sizes, in RunR3()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 530 absl::Span<const int64> slice_sizes) { in DynamicSlice()
757 absl::Span<const int64> slice_sizes) { in Gather()
|
| D | xla_client.py | 1353 def DynamicSlice(self, operand, start_indices, slice_sizes): argument
1775 def Gather(self, a, start_indices, dimension_numbers, slice_sizes): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 759 absl::Span<const int64> slice_sizes) { in DynamicSlice()
782 absl::Span<const int64> slice_sizes) { in DynamicSlice()
1842 absl::Span<const int64> slice_sizes) { in Gather()
2882 absl::Span<const int64> slice_sizes) { in DynamicSlice()
2886 absl::Span<const int64> slice_sizes) { in DynamicSlice()
3426 absl::Span<const int64> slice_sizes) { in Gather()
|