Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_computation.h571 CHECK(!IsCustomCallComputation() && !IsAsyncComputation()); in SetFusionInstruction()
577 bool IsCustomCallComputation() const { return is_custom_call_computation_; } in IsCustomCallComputation() function
606 CHECK(!IsFusionComputation() && !IsCustomCallComputation()); in AddAsyncInstruction()
625 return IsFusionComputation() || IsCustomCallComputation(); in IsCalledComputation()
H A Dhlo_module_test.cc208 EXPECT_TRUE(cloned_computation->IsCustomCallComputation()); in TEST_F()
248 EXPECT_TRUE(cloned_computation_0->IsCustomCallComputation()); in TEST_F()
250 EXPECT_TRUE(cloned_computation_1->IsCustomCallComputation()); in TEST_F()
H A Dhlo_instructions.cc256 CHECK(!async_computation->IsCustomCallComputation()); in HloAsyncInstruction()
271 CHECK(!async_computation->IsCustomCallComputation()); in HloAsyncInstruction()