Searched refs:PER_SHARD_V1 (Results 1 – 3 of 3) sorted by relevance
40 PER_SHARD_V1 = 1 variable in InputPipelineConfig165 per_host_input_for_training = InputPipelineConfig.PER_SHARD_V1
399 tpu_config.InputPipelineConfig.PER_SHARD_V1))
2321 tpu_config.InputPipelineConfig.PER_SHARD_V1 and