Searched defs:limit_indices (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 42 std::vector<int64> limit_indices(input_dims.begin(), input_dims.end()); in TopK() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unpack_op.cc | 68 std::vector<int64> limit_indices(input_shape.dims()); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 95 std::vector<int64> limit_indices(input->shape().dimensions().begin(), in MaybePaddedAndSlicedInput() local
349 std::vector<int64> limit_indices( in CanonicalizeBackwardInputConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 72 absl::Span<const int64> limit_indices, in MakeSliceHlo()
|
| D | hlo_evaluator.cc | 1470 std::vector<int64> limit_indices(indices.begin(), indices.end()); in HandleSort() local
|
| D | algebraic_simplifier.cc | 1464 std::array<int64, 2> limit_indices; in OptimizeDotOfConcatHelper() local
|
| D | hlo_instruction.cc | 987 absl::Span<const int64> limit_indices, absl::Span<const int64> strides) { in CreateSlice()
|
| D | hlo_instructions.cc | 966 absl::Span<const int64> limit_indices, in HloSliceInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 517 absl::Span<const int64> limit_indices, in Slice()
|
| D | xla_client.py | 1317 def Slice(self, operand, start_indices, limit_indices, strides=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 723 absl::Span<const int64> limit_indices, in Slice()
2869 absl::Span<const int64> limit_indices, in Slice()
|