| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 126 std::vector<int64_t> end_indices = reshaped_permuted_shape; in BatchToSpace() local
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | example_proto_fast_parsing.cc | 31 const std::vector<size_t>& end_indices = in CountSparseFeatures() local
|
| D | parse_example.cc | 90 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
395 const std::vector<size_t>& end_indices = sparse_buffer.example_end_indices; in CountSparseFeatures() local
545 std::vector<size_t>& end_indices = in FastParseExampleLite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 111 std::vector<int64_t> end_indices(result_shape.dimensions().begin(), in PadConv() local
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| D | bert_question_answerer.cc | 313 auto end_indices = ReverseSortIndices(end_logits); in Postprocess() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 553 end_indices(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
858 end_indices(batch_space_collapsed_reshape->shape().dimensions().begin(), in ChangeSpatialSizeOnSpaceToBatchedShape() local
2138 end_indices(new_consumer->shape().dimensions().begin(), in Propagate() local
2248 end_indices(previous_shape.dimensions().begin(), in Propagate() local
2421 end_indices(new_dimensions.begin(), new_dimensions.end()), in BatchToSpace() local
|
| D | sharding_propagation.cc | 1366 std::vector<int64_t> end_indices = tile_assignment.dimensions(); in GetShardingFromUser() local
|
| D | algebraic_simplifier.cc | 3866 std::vector<int64_t> end_indices; in HandlePad() local
6238 std::vector<int64_t> end_indices; in HandleTranspose() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3139 auto end_indices = rewriter.create<tensor::FromElementsOp>( in matchAndRewrite() local
3347 auto end_indices = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
3413 SmallVector<Value, 4> end_indices; in matchAndRewrite() local
3546 auto end_indices = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
3593 ArrayRef<int64_t> end_indices, in rewriteWithConstantBegin()
3807 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
3839 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
5516 auto end_indices = llvm::to_vector<4>(value_type.getShape()); in matchAndRewrite() local
5565 SmallVector<Value, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1032 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
1094 const std::vector<size_t>& end_indices = in CountSparseFeatures() local
1380 std::vector<size_t>& end_indices = in FastParseExample() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 2102 absl::InlinedVector<int64_t, 4> end_indices(/*n=*/shape().rank(), 1); in IsAllFirst() local
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 4082 def load(prefix, increments, start_indices, end_indices): argument
|