Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_ops.py282 def _define_full_covariance_probs(self, shard_id, shard): argument
303 def _define_diag_covariance_probs(self, shard_id, shard): argument
327 def _define_log_prob_operation(self, shard_id, shard): argument
344 def _define_prior_log_prob_operation(self, shard_id): argument
356 def _define_expectation_operation(self, shard_id): argument
369 def _define_partial_maximization_operation(self, shard_id, shard): argument
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dnaming.cc28 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards) { in DataFilename()
/external/tensorflow/tensorflow/core/protobuf/
Dtensor_bundle.proto49 int32 shard_id = 3; field
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_filter_ops.cc465 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
Dconv_grad_input_ops.cc536 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local