Searched refs:dynamic_slice_sizes (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo.proto | 90 repeated int64 dynamic_slice_sizes = 20;
|
H A D | hlo_instruction.h | 1034 const std::vector<int64>& dynamic_slice_sizes() const { function in class:xla::HloInstruction
|
H A D | hlo_verifier.cc | 230 dynamic_slice->dynamic_slice_sizes()));
|
H A D | hlo_instruction.cc | 133 for (int64 dynamic_slice_size : proto.dynamic_slice_sizes()) { 2201 StrCat("dynamic_slice_sizes={", Join(dynamic_slice_sizes(), ","), "}"));
|
H A D | hlo_evaluator.cc | 1236 dynamic_slice->dynamic_slice_sizes()));
|
/external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
H A D | hlo_parser.cc | 811 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 D | hlo_parser_test.cc | 519 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