Home
last modified time | relevance | path

Searched refs:CreateIota (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc158 HloInstruction::CreateIota(mask_shape, dim)); in Run()
Dstable_sort_expander.cc76 HloInstruction::CreateIota(iota_shape, sort->sort_dimension())); in ExpandInstruction()
Dalgebraic_simplifier_test.cc2407 auto iota = builder.AddInstruction(HloInstruction::CreateIota( in TEST_F()
2429 HloInstruction::CreateIota(ShapeUtil::MakeShape(F32, {1, 1}), 0)); in TEST_F()
2450 HloInstruction::CreateIota(ShapeUtil::MakeShape(F32, {3, 2}), 1)); in TEST_F()
2470 HloInstruction::CreateIota(ShapeUtil::MakeShape(F32, {3, 2, 4}), 2)); in TEST_F()
2492 HloInstruction::CreateIota(ShapeUtil::MakeShape(F32, {3, 2, 2}), 2)); in TEST_F()
2515 HloInstruction::CreateIota(ShapeUtil::MakeShape(F32, {3, 2, 4, 2}), 2)); in TEST_F()
3929 builder.AddInstruction(HloInstruction::CreateIota(r2f32, 1)); in TEST_F()
3949 builder.AddInstruction(HloInstruction::CreateIota(r3f32, 1)); in TEST_F()
Dpattern_matcher_test.cc685 SetName("i", HloInstruction::CreateIota(ShapeUtil::MakeShape(S32, {42}), in TEST()
Dhlo_instruction.h376 static std::unique_ptr<HloInstruction> CreateIota(const Shape& shape,
Dalgebraic_simplifier.cc822 constant, HloInstruction::CreateIota(constant->shape(), 0)); in HandleConstant()
2122 HloInstruction::CreateIota( in HandleBroadcast()
2645 HloInstruction::CreateIota(reshape->shape(), opt_dims->front())); in HandleReshape()
Dhlo_instruction.cc553 instruction = CreateIota(shape, proto.dimensions(0)); in CreateFromProto()
653 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateIota( in CreateIota() function in xla::HloInstruction
Dhlo_parser.cc727 HloInstruction::CreateIota(shape, *iota_dimension)); in ParseInstructionRhs()