Home
last modified time | relevance | path

Searched refs:CreateComputationWithSignature (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_creation_utils_test.cc41 CreateComputationWithSignature({&input_shape}, output_shape, "entry") in CreateModuleWithProgramShape()
56 CreateComputationWithSignature({&input_shape}, output_shape, "entry") in CreateModuleWithProgramShape()
351 CreateComputationWithSignature({&input_shape0, &input_shape1}, in TEST_F()
H A Dwhile_util.cc163 CreateComputationWithSignature( in MakeCountedLoopConditionComputation()
186 CreateComputationWithSignature( in MakeCountedLoopBodyComputation()
H A Dhlo_creation_utils.h358 StatusOr<std::unique_ptr<HloComputation>> CreateComputationWithSignature(
H A Dhlo_creation_utils.cc711 StatusOr<std::unique_ptr<HloComputation>> CreateComputationWithSignature( in CreateComputationWithSignature() function