Home
last modified time | relevance | path

Searched refs:pad_to_static (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder_test.cc189 HloInstruction* pad_to_static = custom_call_2->users()[0]; in TEST_F() local
190 ASSERT_THAT(pad_to_static->opcode(), HloOpcode::kCustomCall); in TEST_F()
191 ASSERT_THAT(pad_to_static->custom_call_target(), "PadToStatic"); in TEST_F()
Ddynamic_padder.cc1493 HloInstruction* pad_to_static = in InsertPadToStaticOnInstruction() local
1498 data_output_shape, pad_to_static, 0)); in InsertPadToStaticOnInstruction()
1675 HloInstruction* pad_to_static = in ConvertToStatic() local
1680 data_output_shape, pad_to_static, 0)); in ConvertToStatic()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc816 auto pad_to_static = mlir::cast<mlir::lmhlo::CustomCallOp>(mlir_input.op); in EmitPadToStaticFromMlir() local
818 std::string ir_name = mlir::GetNameFromLoc(pad_to_static.getLoc()); in EmitPadToStaticFromMlir()
823 BuildKernelThunkForMlir(pad_to_static, mlir_input.thunk_info, in EmitPadToStaticFromMlir()
835 TypeToShape(pad_to_static.output().front().getType()); in EmitPadToStaticFromMlir()
837 TypeToShape(pad_to_static.args().front().getType()); in EmitPadToStaticFromMlir()
852 for (int64 i = 1; i < pad_to_static.output().size(); ++i) { in EmitPadToStaticFromMlir()
855 const Shape& dim_shape = TypeToShape(pad_to_static.output()[i].getType()); in EmitPadToStaticFromMlir()
875 for (int64 i = 1; i < pad_to_static.output().size(); ++i) { in EmitPadToStaticFromMlir()
933 pad_to_static, launch_dimensions.launch_bound(), &b_))); in EmitPadToStaticFromMlir()