Home
last modified time | relevance | path

Searched defs:shard_id (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_optimization_parameters_utils.cc427 int shard_id; in operator ()() local
457 int shard_id; in operator ()() local
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dnaming.cc28 string DataFilename(StringPiece prefix, int32_t shard_id, int32_t num_shards) { in DataFilename()
/external/tensorflow/tensorflow/core/protobuf/
Dtensor_bundle.proto51 int32 shard_id = 3; field
/external/libtextclassifier/native/utils/grammar/utils/
Dlocale-shard-map.cc94 int shard_id = shard_to_locale_data_.size(); in AddLocalTags() local
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_evaluation_test.py167 def eval_fn(total_shard, shard_id, metric): argument
/external/angle/scripts/
Dprocess_angle_perf_results.py476 def _find_device_id_by_shard_id(benchmarks_shard_map_file, shard_id): argument
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops_3d.cc593 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
1062 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
Dconv_grad_filter_ops.cc586 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
/external/tensorflow/tensorflow/core/data/
Dsnapshot_utils.cc84 std::string ShardDirectory(const std::string& run_directory, int64_t shard_id) { in ShardDirectory()