Searched refs:pad_to_static (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dynamic_padder_test.cc | 189 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()
|
D | dynamic_padder.cc | 1493 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/ |
D | ir_emitter_unnested.cc | 816 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()
|