Home
last modified time | relevance | path

Searched refs:pipeline_execution_with_tensor_core (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.estimator.tpu.experimental.-embedding-config-spec.pbtxt31 name: "pipeline_execution_with_tensor_core"
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt13 …pec: "args=[\'self\', \'feature_config\', \'optimizer\', \'pipeline_execution_with_tensor_core\'],…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt13 …pec: "args=[\'self\', \'feature_config\', \'optimizer\', \'pipeline_execution_with_tensor_core\'],…
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_v2.py240 pipeline_execution_with_tensor_core: bool = False):
274 pipeline_execution_with_tensor_core)
609 config_proto.pipeline_execution_with_tensor_core = (
H A Dtpu_embedding.py1309 pipeline_execution_with_tensor_core=False, argument
1447 pipeline_execution_with_tensor_core)
1608 config_proto.pipeline_execution_with_tensor_core = (
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/
H A Dtpu_embedding_configuration.proto91 bool pipeline_execution_with_tensor_core = 7; field