Home
last modified time | relevance | path

Searched defs:end_indices (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatchtospace_op.cc126 std::vector<int64> end_indices = reshaped_permuted_shape; in BatchToSpace() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc102 std::vector<int64> end_indices(result_shape.dimensions().begin(), in PadConv() local
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc929 const auto& end_indices = buffer.example_end_indices; in FillAndCopyVarLen() local
1085 const std::vector<size_t>& end_indices = in FastParseExample() local
1165 std::vector<size_t>& end_indices = in FastParseExample() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2274 std::vector<int64> end_indices; in HandlePad() local