Searched refs:slice_to_dynamic (Results 1 – 2 of 2) sorted by relevance
185 HloInstruction* slice_to_dynamic = custom_call_1->mutable_operand(0); in TEST_F() local186 ASSERT_THAT(slice_to_dynamic->opcode(), HloOpcode::kCustomCall); in TEST_F()187 ASSERT_THAT(slice_to_dynamic->custom_call_target(), "SliceToDynamic"); in TEST_F()192 slice_to_dynamic = module_->entry_computation()->root_instruction(); in TEST_F()193 ASSERT_THAT(slice_to_dynamic->opcode(), HloOpcode::kCustomCall); in TEST_F()194 ASSERT_THAT(slice_to_dynamic->custom_call_target(), "SliceToDynamic"); in TEST_F()
943 auto slice_to_dynamic = mlir::cast<mlir::lmhlo::CustomCallOp>(mlir_input.op); in EmitSliceToDynamicFromMlir() local945 std::string ir_name = mlir::GetNameFromLoc(slice_to_dynamic.getLoc()); in EmitSliceToDynamicFromMlir()950 BuildKernelThunkForMlir(slice_to_dynamic, mlir_input.thunk_info, in EmitSliceToDynamicFromMlir()954 TypeToShape(slice_to_dynamic.args().front().getType()); in EmitSliceToDynamicFromMlir()955 TF_RET_CHECK(slice_to_dynamic.output().size() == 1); in EmitSliceToDynamicFromMlir()957 TypeToShape(slice_to_dynamic.output().front().getType()); in EmitSliceToDynamicFromMlir()980 for (int64 i = 1; i < slice_to_dynamic.args().size(); ++i) { in EmitSliceToDynamicFromMlir()995 for (int64 i = 1; i < slice_to_dynamic.args().size(); ++i) { in EmitSliceToDynamicFromMlir()1056 slice_to_dynamic, in EmitSliceToDynamicFromMlir()