Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dcreate_dtensor_mlir_passes.h85 CreateDTensorPropagateDeviceIdToFunctionArgs();
H A Dpropagate_device_id_to_function_args.cc187 CreateDTensorPropagateDeviceIdToFunctionArgs() { in CreateDTensorPropagateDeviceIdToFunctionArgs() function
H A Ddtensor_mlir_passes.cc97 pm->addPass(CreateDTensorPropagateDeviceIdToFunctionArgs()); in CreateDTensorMLIRPass()
H A DPasses.td200 let constructor = "CreateDTensorPropagateDeviceIdToFunctionArgs()";