Searched defs:limit_indices (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 77 std::vector<int64_t> limit_indices(input_dims.begin(), input_dims.end()); in TopK() local
204 std::vector<int64_t> limit_indices(input_dims.begin(), input_dims.end()); in TopKWithPartitions() local
219 std::vector<int64_t> limit_indices(input_dims.begin(), input_dims.end()); in TopKWithPartitions() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unpack_op.cc | 68 std::vector<int64_t> limit_indices(input_shape.dims()); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 97 std::vector<int64_t> limit_indices(input->shape().dimensions().begin(), in MaybePaddedAndSlicedInput() local
356 std::vector<int64_t> limit_indices( in CanonicalizeBackwardInputConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_merger.cc | 198 DimensionVector limit_indices(new_dot_shape.dimensions().begin(), in TryMergeSameOperand() local
|
| D | hlo_creation_utils.cc | 104 absl::Span<const int64_t> limit_indices, in MakeSliceHlo()
|
| D | hlo_evaluator.cc | 3494 std::vector<int64_t> limit_indices(indices.begin(), indices.end()); in HandleSort() local
|
| D | hlo_instruction.cc | 1507 absl::Span<const int64_t> limit_indices, in CreateSlice()
|
| D | hlo_instructions.cc | 1390 absl::Span<const int64_t> limit_indices, absl::Span<const int64_t> strides) in HloSliceInstruction()
|
| D | algebraic_simplifier.cc | 2226 std::array<int64_t, 2> limit_indices; in OptimizeDotOfConcatHelper() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 794 std::vector<int64_t> limit_indices(hlo->shape().dimensions().begin(), in ExchangeHalo() local
1228 std::vector<int64_t> limit_indices(hlo_dims.begin(), hlo_dims.end()); in SliceFirstK() local
|
| D | spmd_partitioner.cc | 2419 std::vector<int64_t> limit_indices(rank); in HandleSlice() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 654 absl::Span<const int64_t> limit_indices, in SliceInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1019 absl::Span<const int64_t> limit_indices, in Slice()
1032 absl::Span<const int64_t> limit_indices, in SliceInternal()
4111 absl::Span<const int64_t> limit_indices, in Slice()
|