Searched refs:sharding_enabled (Results 1 – 1 of 1) sorted by relevance
2398 bool sharding_enabled = (arg_shape.handle_type != DT_COMPLEX64 && in BuildCompileNode() local2403 sharding_enabled ? tpu::TPUCompileMetadataProto::Arg::TENTATIVE in BuildCompileNode()