Searched defs:limit_indices (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 47 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 | gpu_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/mlir/xla/ir/ |
| D | hlo_ops.cc | 969 DenseIntElementsAttr limit_indices, in build()
990 DenseIntElementsAttr limit_indices, in InferOutputTypes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 73 absl::Span<const int64> limit_indices, in MakeSliceHlo()
|
| D | algebraic_simplifier.cc | 1462 std::array<int64, 2> limit_indices; in OptimizeDotOfConcatHelper() local
|
| D | hlo_evaluator.cc | 2170 std::vector<int64> limit_indices(indices.begin(), indices.end()); in HandleSort() local
|
| D | hlo_instruction.cc | 1074 absl::Span<const int64> limit_indices, absl::Span<const int64> strides) { in CreateSlice()
|
| D | hlo_instructions.cc | 1060 absl::Span<const int64> limit_indices, in HloSliceInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1109 def Slice(self, operand, start_indices, limit_indices, strides=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 843 absl::Span<const int64> limit_indices, in Slice()
2934 absl::Span<const int64> limit_indices, in Slice()
|