Searched refs:CreateDynamicSlice (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_index_splitter.cc | 87 : HloInstruction::CreateDynamicSlice( in Run()
|
| D | hlo_creation_utils.cc | 144 return computation->AddInstruction(HloInstruction::CreateDynamicSlice( in MakeDynamicSliceHlo()
|
| D | hlo_instruction.h | 584 static std::unique_ptr<HloInstruction> CreateDynamicSlice(
|
| D | hlo_evaluator_test.cc | 1822 HloInstruction::CreateDynamicSlice(shape, operand, {zero, one}, {2, 3})); in TEST_P()
1860 HloInstruction::CreateDynamicSlice(shape, operand, {two, one}, {2, 3})); in TEST_P()
|
| D | algebraic_simplifier_test.cc | 3820 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
3853 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
4659 auto* ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_P()
4740 auto* ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_P()
|
| D | hlo_instruction.cc | 497 instruction = CreateDynamicSlice( in CreateFromProto()
992 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateDynamicSlice( in CreateDynamicSlice() function in xla::HloInstruction
|
| D | hlo_dataflow_analysis_test.cc | 2186 auto ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
|
| D | algebraic_simplifier.cc | 1624 computation_->AddInstruction(HloInstruction::CreateDynamicSlice( in OptimizeDotOfGather()
|
| D | hlo_parser.cc | 1278 instruction = builder->AddInstruction(HloInstruction::CreateDynamicSlice( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 350 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
394 HloInstruction::CreateDynamicSlice(result_shape, param0, {param1}, {2})); in TEST_F()
564 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
599 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
608 auto slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 544 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in XLA_TEST_F()
|