Searched refs:dynamic_slice_sizes (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter.cc | 90 dynamic_op->dynamic_slice_sizes()); in Run()
|
| D | hlo.proto | 102 repeated int64 dynamic_slice_sizes = 20; field
|
| D | hlo_parser_test.cc | 603 …ice(s32[2,2,258]{2,1,0} %original_parameter, s32[3]{0} %concatenate), dynamic_slice_sizes={2,2,258} in CreateTestCases()
617 …nal_parameter, s32[] %constant, s32[] %constant, s32[] %start_index), dynamic_slice_sizes={2,2,258} in CreateTestCases()
|
| D | hlo_instruction.cc | 484 absl::c_copy(proto.dynamic_slice_sizes(), slice_sizes.begin()); in CreateFromProto()
3512 const std::vector<int64>& HloInstruction::dynamic_slice_sizes() const { in dynamic_slice_sizes() function in xla::HloInstruction
3513 return Cast<HloDynamicSliceInstruction>(this)->dynamic_slice_sizes(); in dynamic_slice_sizes()
|
| D | hlo_instruction.h | 1590 const std::vector<int64>& dynamic_slice_sizes() const;
|
| D | hlo_parser.cc | 1264 optional<std::vector<int64>> dynamic_slice_sizes; in ParseInstructionRhs() local
1266 /*required=*/true, AttrTy::kBracedInt64List, &dynamic_slice_sizes}; in ParseInstructionRhs()
1281 *dynamic_slice_sizes)); in ParseInstructionRhs()
|
| D | hlo_verifier.cc | 581 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
|
| D | hlo_instructions.h | 1321 const std::vector<int64>& dynamic_slice_sizes() const { in dynamic_slice_sizes() function
|
| D | hlo_instructions.cc | 2284 return {StrCat("dynamic_slice_sizes={", StrJoin(dynamic_slice_sizes(), ","), in ExtraAttributesToStringImpl()
|
| D | hlo_evaluator_typed_visitor.h | 1465 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
|