Searched defs:end_indices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 126 std::vector<int64> end_indices = reshaped_permuted_shape; in BatchToSpace() local
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | example_proto_fast_parsing.cc | 28 const std::vector<size_t>& end_indices = in CountSparseFeatures() local
|
| D | parse_example.cc | 88 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
393 const std::vector<size_t>& end_indices = sparse_buffer.example_end_indices; in CountSparseFeatures() local
543 std::vector<size_t>& end_indices = in FastParseExampleLite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 107 std::vector<int64> 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 | 389 end_indices(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
1501 end_indices(new_consumer->shape().dimensions().begin(), in Propagate() local
1611 end_indices(previous_shape.dimensions().begin(), in Propagate() local
1741 end_indices(new_dimensions.begin(), new_dimensions.end()), in BatchToSpace() local
|
| D | sharding_propagation.cc | 1190 std::vector<int64> end_indices = tile_assignment.dimensions(); in GetShardingFromUser() local
|
| D | algebraic_simplifier.cc | 3488 std::vector<int64> end_indices; in HandlePad() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1016 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
1078 const std::vector<size_t>& end_indices = in CountSparseFeatures() local
1364 std::vector<size_t>& end_indices = in FastParseExample() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2942 auto end_indices = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
3054 auto end_indices = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
3101 ArrayRef<int64_t> end_indices, in rewriteWithConstantBegin()
3318 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
3350 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
4775 auto end_indices = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
4820 auto end_indices = llvm::to_vector<4>(value_type.getShape()); in matchAndRewrite() local
|