| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | example_proto_fast_parsing.cc | 28 const std::vector<size_t>& end_indices = in CountSparseFeatures() local
30 *total_num_features += end_indices.back(); in CountSparseFeatures()
31 *max_num_features = std::max(*max_num_features, end_indices[0]); in CountSparseFeatures()
32 for (size_t i = 1; i < end_indices.size(); ++i) { in CountSparseFeatures()
33 size_t example_size = end_indices[i] - end_indices[i - 1]; in CountSparseFeatures()
|
| D | parse_example.cc | 88 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
89 const size_t examples_in_buffer = end_indices.size(); in FillAndCopyVarLen()
98 const size_t num_elems = end_indices[j] - elements_tally; in FillAndCopyVarLen()
104 elements_tally = end_indices[j]; in FillAndCopyVarLen()
393 const std::vector<size_t>& end_indices = sparse_buffer.example_end_indices; in CountSparseFeatures() local
394 *total_num_features += end_indices.back(); in CountSparseFeatures()
395 *max_num_features = std::max(*max_num_features, end_indices[0]); in CountSparseFeatures()
396 for (size_t i = 1; i < end_indices.size(); ++i) { in CountSparseFeatures()
397 size_t example_size = end_indices[i] - end_indices[i - 1]; in CountSparseFeatures()
543 std::vector<size_t>& end_indices = in FastParseExampleLite() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 126 std::vector<int64> end_indices = reshaped_permuted_shape; in BatchToSpace() local
134 end_indices[1 + i] -= crop_end; in BatchToSpace()
136 ctx, start_indices[1 + i] <= end_indices[1 + i], in BatchToSpace()
142 xla::Slice(reshaped_permuted, start_indices, end_indices, strides); in BatchToSpace()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1016 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
1017 const size_t examples_in_buffer = end_indices.size(); in FillAndCopyVarLen()
1027 const size_t num_elems = end_indices[j] - elements_tally; in FillAndCopyVarLen()
1033 elements_tally = end_indices[j]; in FillAndCopyVarLen()
1078 const std::vector<size_t>& end_indices = in CountSparseFeatures() local
1080 *total_num_features += end_indices.back(); in CountSparseFeatures()
1081 *max_num_features = std::max(*max_num_features, end_indices[0]); in CountSparseFeatures()
1082 for (size_t i = 1; i < end_indices.size(); ++i) { in CountSparseFeatures()
1083 size_t example_size = end_indices[i] - end_indices[i - 1]; in CountSparseFeatures()
1364 std::vector<size_t>& end_indices = in FastParseExample() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 389 end_indices(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
393 end_indices[activations_batch_dim] = batch_size - 1; in HaloDuplicateWithSlice()
394 end_indices[spatial_dimension_to_split] = spatial_split_size; in HaloDuplicateWithSlice()
397 end_indices, strides)); in HaloDuplicateWithSlice()
1501 end_indices(new_consumer->shape().dimensions().begin(), in Propagate() local
1505 end_indices[new_space_dim] = new_space_size + halo_size; in Propagate()
1506 end_indices[new_batch_dim] = batch_size - 1; in Propagate()
1511 MakeSliceHlo(new_consumer, start_indices, end_indices, strides)); in Propagate()
1611 end_indices(previous_shape.dimensions().begin(), in Propagate() local
1617 MakeSliceHlo(new_consumer, start_indices, end_indices, strides)); in Propagate()
[all …]
|
| D | sharding_propagation.cc | 1190 std::vector<int64> end_indices = tile_assignment.dimensions(); in GetShardingFromUser() local
1192 end_indices[cdim] = CeilOfRatio( in GetShardingFromUser()
1195 tile_assignment.Slice(start_indices, end_indices); in GetShardingFromUser()
|
| D | algebraic_simplifier.cc | 3488 std::vector<int64> end_indices; in HandlePad() local
3502 end_indices.push_back(end); in HandlePad()
3508 MakeSliceHlo(nonzero_pad, start_indices, end_indices, strides)); in HandlePad()
|
| /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
116 result_shape, new_conv_result, start_indices, end_indices, strides)); in PadConv()
|
| /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
319 int end = end_indices[end_index]; in Postprocess()
|
| /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
2951 end_indices[dim_index] = (i + 1) * slice_size; in matchAndRewrite()
2955 GetI64ElementsAttr(end_indices, &rewriter), in matchAndRewrite()
3054 auto end_indices = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
3062 end_indices[dim_index] = begin_indices[dim_index] + split_sizes[i]; in matchAndRewrite()
3065 GetI64ElementsAttr(end_indices, &rewriter), in matchAndRewrite()
3068 begin_indices[dim_index] = end_indices[dim_index]; in matchAndRewrite()
3101 ArrayRef<int64_t> end_indices, in rewriteWithConstantBegin() argument
3120 int64_t end = end_indices[i]; in rewriteWithConstantBegin()
3318 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 15402 // `begin_indices`, `end_indices`, and `strides` with their canonical
15451 // `begin_indices`, `end_indices`, and `strides` with their canonical
|