Searched refs:kDynamicSlice (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter_test.cc | 33 const char* const kDynamicSlice = R"( in TEST_F() local
48 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kDynamicSlice, config)); in TEST_F()
107 const char* const kDynamicSlice = R"( in TEST_F() local
124 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kDynamicSlice, config)); in TEST_F()
|
| D | hlo_opcode_test.cc | 59 case HloOpcode::kDynamicSlice: in TEST()
|
| D | dynamic_index_splitter.cc | 40 case HloOpcode::kDynamicSlice: in Run()
|
| D | bfloat16_support.cc | 104 case HloOpcode::kDynamicSlice: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | hlo_opcode.h | 83 V(kDynamicSlice, "dynamic-slice", kHloOpcodeIsVariadic) \
|
| D | instruction_fusion.cc | 74 case HloOpcode::kDynamicSlice: in IsExpensive()
|
| D | hlo_instruction.cc | 482 case HloOpcode::kDynamicSlice: { in CreateFromProto()
1382 case HloOpcode::kDynamicSlice: in CloneWithNewOperands()
1834 case HloOpcode::kDynamicSlice: in IdenticalSlowPath()
2619 case HloOpcode::kDynamicSlice: in Visit()
|
| D | algebraic_simplifier.cc | 1521 if (a->opcode() != HloOpcode::kDynamicSlice) { in OptimizeDotOfGather()
1577 bool lhs_is_dynamic_slice = lhs->opcode() == HloOpcode::kDynamicSlice; in OptimizeDotOfGather()
2025 case HloOpcode::kDynamicSlice: in OutputIsSubsetOfOperandElements()
|
| D | hlo_graph_dumper.cc | 991 case HloOpcode::kDynamicSlice: in GetInstructionColor()
|
| D | hlo_instructions.cc | 2236 : HloDynamicIndexInstruction(HloOpcode::kDynamicSlice, shape), in HloDynamicSliceInstruction()
2246 : HloDynamicIndexInstruction(HloOpcode::kDynamicSlice, shape), in HloDynamicSliceInstruction()
|
| D | layout_assignment.cc | 2000 case HloOpcode::kDynamicSlice: in InstructionCanChangeLayout()
|
| D | elemental_ir_emitter.cc | 2374 case HloOpcode::kDynamicSlice: in MakeElementGenerator()
|
| D | algebraic_simplifier_test.cc | 4689 HloOpcode::kDynamicSlice); in TEST_P()
4762 HloOpcode::kDynamicSlice); in TEST_P()
|
| D | hlo_parser.cc | 1263 case HloOpcode::kDynamicSlice: { in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 360 {HloOpcode::kTanh, HloOpcode::kDynamicSlice, HloOpcode::kReshape, in TEST_F()
402 module.get(), {HloOpcode::kNegate, HloOpcode::kDynamicSlice, in TEST_F()
580 {HloOpcode::kDynamicSlice, HloOpcode::kDynamicUpdateSlice, in TEST_F()
624 {HloOpcode::kDynamicSlice, HloOpcode::kDynamicSlice, in TEST_F()
|
| D | cpu_instruction_fusion.cc | 36 hlo.opcode() == HloOpcode::kDynamicSlice || in CanBeLoopFused()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 401 if ((opcode == HloOpcode::kDynamicSlice && op_num >= 1) || in FindConstrainedUses()
447 case HloOpcode::kDynamicSlice: in CreateLiteralForConstrainedUses()
450 const Shape& slice_shape = use->opcode() == HloOpcode::kDynamicSlice in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 167 instr.opcode() == HloOpcode::kDynamicSlice || in IsLoopFusible()
|
| D | fusion_merger.cc | 73 instruction->opcode() == HloOpcode::kDynamicSlice; in CalculateBytesReadByFusionParameter()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 775 return AddInstruction(std::move(instr), HloOpcode::kDynamicSlice, in DynamicSlice()
804 return AddInstruction(std::move(instr), HloOpcode::kDynamicSlice, operands); in DynamicSlice()
|