/external/tensorflow/tensorflow/compiler/xla/service/ |
hlo.proto | 90 repeated int64 dynamic_slice_sizes = 20;
|
hlo_instruction.h | 1034 const std::vector<int64>& dynamic_slice_sizes() const { function in class:xla::HloInstruction [all...] |
hlo_verifier.cc | 230 dynamic_slice->dynamic_slice_sizes())); [all...] |
hlo_instruction.cc | 133 for (int64 dynamic_slice_size : proto.dynamic_slice_sizes()) { [all...] |
hlo_evaluator.cc | [all...] |
/external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
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)); [all...] |
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} [all...] |