Searched refs:dynamic_slice_sizes (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo.proto90 repeated int64 dynamic_slice_sizes = 20;
H A Dhlo_instruction.h1034 const std::vector<int64>& dynamic_slice_sizes() const { function in class:xla::HloInstruction
H A Dhlo_verifier.cc230 dynamic_slice->dynamic_slice_sizes()));
H A Dhlo_instruction.cc133 for (int64 dynamic_slice_size : proto.dynamic_slice_sizes()) {
2201 StrCat("dynamic_slice_sizes={", Join(dynamic_slice_sizes(), ","), "}"));
H A Dhlo_evaluator.cc1236 dynamic_slice->dynamic_slice_sizes()));
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
H A Dhlo_parser.cc811 optional<std::vector<int64>> dynamic_slice_sizes; local
812 attrs["dynamic_slice_sizes"] = {
813 /*required=*/true, AttrTy::kBracedInt64List, &dynamic_slice_sizes};
820 *dynamic_slice_sizes));
H A Dhlo_parser_test.cc519 ROOT %dynamic-slice = s32[2,2,258]{2,1,0} dynamic-slice(s32[2,2,258]{2,1,0} %original_parameter, s32[3]{0} %concatenate), dynamic_slice_sizes={2,2,258}

Completed in 173 milliseconds