Home
last modified time | relevance | path

Searched refs:HloDynamicSliceInstruction (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc2233 HloDynamicSliceInstruction::HloDynamicSliceInstruction( in HloDynamicSliceInstruction() function in xla::HloDynamicSliceInstruction
2242 HloDynamicSliceInstruction::HloDynamicSliceInstruction( in HloDynamicSliceInstruction() function in xla::HloDynamicSliceInstruction
2274 HloInstructionProto HloDynamicSliceInstruction::ToProto() const { in ToProto()
2282 std::vector<string> HloDynamicSliceInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
2288 bool HloDynamicSliceInstruction::IdenticalSlowPath( in IdenticalSlowPath()
2296 HloDynamicSliceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()
2301 return absl::make_unique<HloDynamicSliceInstruction>( in CloneWithNewOperandsImpl()
2304 return absl::make_unique<HloDynamicSliceInstruction>( in CloneWithNewOperandsImpl()
Dhlo_instructions.h1305 class HloDynamicSliceInstruction : public HloDynamicIndexInstruction {
1307 explicit HloDynamicSliceInstruction(const Shape& shape,
1311 explicit HloDynamicSliceInstruction(
Dhlo_dataflow_analysis.cc1012 auto* ds = DynCast<HloDynamicSliceInstruction>(user); in CanDoInPlaceDynamicUpdateSlice()
Dhlo_instruction.cc996 return absl::make_unique<HloDynamicSliceInstruction>( in CreateDynamicSlice()
3509 return Cast<HloDynamicSliceInstruction>(this)->slice_sizes(dimension); in slice_sizes()
3513 return Cast<HloDynamicSliceInstruction>(this)->dynamic_slice_sizes(); in dynamic_slice_sizes()
Dalgebraic_simplifier.cc1578 HloDynamicSliceInstruction* dynamic_slice = in OptimizeDotOfGather()
1579 lhs_is_dynamic_slice ? Cast<HloDynamicSliceInstruction>(lhs) in OptimizeDotOfGather()
1580 : Cast<HloDynamicSliceInstruction>(rhs); in OptimizeDotOfGather()
Dhlo_verifier.cc580 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(), in HandleDynamicSlice()
Dhlo_evaluator_typed_visitor.h1464 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(), in HandleDynamicSlice()