| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter.cc | 41 case HloOpcode::kDynamicUpdateSlice: in Run()
47 bool is_update = dynamic_op->opcode() == HloOpcode::kDynamicUpdateSlice; in Run()
|
| D | dynamic_index_splitter_test.cc | 70 const char* const kDynamicUpdateSlice = R"( in TEST_F() local
87 auto module, ParseAndReturnVerifiedModule(kDynamicUpdateSlice, config)); in TEST_F()
|
| D | hlo_opcode_test.cc | 60 case HloOpcode::kDynamicUpdateSlice: in TEST()
|
| D | bfloat16_support.cc | 107 case HloOpcode::kDynamicUpdateSlice: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | fusion_node_indexing_evaluation.cc | 53 case HloOpcode::kDynamicUpdateSlice: in OpInvalidatesCache()
|
| D | hlo_opcode.h | 85 V(kDynamicUpdateSlice, "dynamic-update-slice", kHloOpcodeIsVariadic) \
|
| D | sharding_propagation.cc | 250 case HloOpcode::kDynamicUpdateSlice: in PickRepresentativeOperand()
306 case HloOpcode::kDynamicUpdateSlice: in SupportSpatialPartitioning()
909 case HloOpcode::kDynamicUpdateSlice: { in InferShardingFromOperands()
937 if (instruction->opcode() != HloOpcode::kDynamicUpdateSlice) { in InferShardingFromOperands()
1212 case HloOpcode::kDynamicUpdateSlice: { in GetShardingFromUser()
1216 if (user.opcode() == HloOpcode::kDynamicUpdateSlice && in GetShardingFromUser()
|
| D | hlo_cost_analysis.cc | 166 case HloOpcode::kDynamicUpdateSlice: in FusionParameterReadBytes()
842 HloOpcode::kDynamicUpdateSlice) { in HandleFusion()
853 ->opcode() == HloOpcode::kDynamicUpdateSlice) { in HandleFusion()
|
| D | hlo_replication_analysis.cc | 155 hlo->opcode() == HloOpcode::kDynamicUpdateSlice || // in DetermineHloInstructionIsReplicated()
|
| D | conditional_simplifier.cc | 121 case HloOpcode::kDynamicUpdateSlice: in TryRemoveConditional()
|
| D | instruction_fusion.cc | 84 case HloOpcode::kDynamicUpdateSlice: in IsExpensive()
|
| D | hlo_dataflow_analysis.cc | 1217 return opcode == HloOpcode::kDynamicUpdateSlice || in IsInPlaceOperation()
1340 if (user->opcode() == HloOpcode::kDynamicUpdateSlice || in CanShareOperandBufferWithUser()
|
| D | hlo_instruction.cc | 645 case HloOpcode::kDynamicUpdateSlice: { in CreateFromProto()
1784 case HloOpcode::kDynamicUpdateSlice: in CloneWithNewOperands()
2136 case HloOpcode::kDynamicUpdateSlice: in IdenticalSlowPath()
2674 if (opcode_ == HloOpcode::kDynamicUpdateSlice) { in IsElementwiseImpl()
3216 case HloOpcode::kDynamicUpdateSlice: in Visit()
3609 case HloOpcode::kDynamicUpdateSlice: in OperandElementUse()
|
| D | dynamic_padder.cc | 129 case HloOpcode::kDynamicUpdateSlice: in ChooseIdentityValue()
1838 if (inst->opcode() == HloOpcode::kDynamicUpdateSlice) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 38 if (instr->opcode() == HloOpcode::kDynamicUpdateSlice) { in MayBeImplementedAsInPlaceDynamicUpdateSlice()
49 return fused_root->opcode() == HloOpcode::kDynamicUpdateSlice && in MayBeImplementedAsInPlaceDynamicUpdateSlice()
59 CHECK_EQ(HloOpcode::kDynamicUpdateSlice, dynamic_update_slice->opcode()); in CanUpdateDynamicSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | multi_output_fusion.cc | 72 HloOpcode::kDynamicUpdateSlice || in LegalToFuse()
75 HloOpcode::kDynamicUpdateSlice)) { in LegalToFuse()
|
| D | gpu_fusible.cc | 223 instr.opcode() == HloOpcode::kDynamicUpdateSlice || in IsLoopFusible()
|
| D | horizontal_loop_fusion.cc | 181 if (root->opcode() == HloOpcode::kDynamicUpdateSlice) { in IsProfitableFusionCandidate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | schedule_aware_all_gather_cse.cc | 46 return operand->opcode() == HloOpcode::kDynamicUpdateSlice && in MayConsiderAsAllGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 35 hlo.opcode() == HloOpcode::kDynamicUpdateSlice || in CanBeLoopFused()
|
| D | parallel_task_assignment.cc | 155 opcode == HloOpcode::kDynamicUpdateSlice || in GetTargetParallelTaskCount()
|
| D | cpu_instruction_fusion_test.cc | 582 {HloOpcode::kDynamicSlice, HloOpcode::kDynamicUpdateSlice, in TEST_F()
627 HloOpcode::kDynamicUpdateSlice, HloOpcode::kReshape, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 511 (opcode == HloOpcode::kDynamicUpdateSlice && op_num >= 2)) { in FindConstrainedUses()
559 case HloOpcode::kDynamicUpdateSlice: { in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 436 return xla::HloOpcode::kDynamicUpdateSlice; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 374 case HloOpcode::kDynamicUpdateSlice: { in ImportInstructionImpl()
|