Searched refs:end_indices (Results 1 – 4 of 4) 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
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 | 929 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
930 const size_t examples_in_buffer = end_indices.size(); in FillAndCopyVarLen()
940 const size_t num_elems = end_indices[j] - elements_tally; in FillAndCopyVarLen()
946 elements_tally = end_indices[j]; in FillAndCopyVarLen()
1085 const std::vector<size_t>& end_indices = in FastParseExample() local
1087 total_num_features += end_indices.back(); in FastParseExample()
1088 max_num_features = std::max(max_num_features, end_indices[0]); in FastParseExample()
1089 for (size_t i = 1; i < end_indices.size(); ++i) { in FastParseExample()
1090 size_t example_size = end_indices[i] - end_indices[i - 1]; in FastParseExample()
1165 std::vector<size_t>& end_indices = in FastParseExample() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores.cc | 102 std::vector<int64> end_indices(result_shape.dimensions().begin(), in PadConv() local
111 result_shape, new_conv_result, start_indices, end_indices, strides)); in PadConv()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2274 std::vector<int64> end_indices; in HandlePad() local
2288 end_indices.push_back(end); in HandlePad()
2294 MakeSliceHlo(nonzero_pad, start_indices, end_indices, strides)); in HandlePad()
|