Home
last modified time | relevance | path

Searched refs:shard_id (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_ops.py271 for shard_id, shard in enumerate(data):
274 self._define_log_prob_operation(shard_id, shard)
275 self._define_prior_log_prob_operation(shard_id)
276 self._define_expectation_operation(shard_id)
277 self._define_partial_maximization_operation(shard_id, shard)
282 def _define_full_covariance_probs(self, shard_id, shard): argument
300 self._probs[shard_id] = (
304 def _define_diag_covariance_probs(self, shard_id, shard): argument
323 self._probs[shard_id] = -0.5 * (
328 def _define_log_prob_operation(self, shard_id, shard): argument
[all …]
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dnaming.cc28 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards) { in DataFilename() argument
30 DCHECK_LT(shard_id, num_shards); in DataFilename()
33 shard_id, num_shards); in DataFilename()
Dnaming.h43 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards);
Dtensor_bundle.cc657 {DataFilename(prefix, to_merge_entry.shard_id(), num_shards), in MergeOneBundle()
826 io::InputBuffer* buffered_file = data_[entry.shard_id()]; in GetValue()
830 DataFilename(prefix_, entry.shard_id(), num_shards_), &file)); in GetValue()
833 data_[entry.shard_id()] = buffered_file; in GetValue()
/external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/
Dimagenet_model_evaluator.cc100 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats, in OnSingleImageEvaluationComplete() argument
104 observer->OnSingleImageEvaluationComplete(shard_id, stats, image); in OnSingleImageEvaluationComplete()
178 Status EvaluateModelForShard(const uint64_t shard_id, in EvaluateModelForShard() argument
220 shard_id, eval->GetTopKAccuracySoFar(), image_label.image); in EvaluateModelForShard()
324 const uint64_t shard_id = i + 1; in EvaluateModel() local
325 shard_id_image_count_map[shard_id] = image_label.size(); in EvaluateModel()
326 auto func = [shard_id, &image_label, &model_labels, this, &observer, &eval, in EvaluateModel()
328 TF_CHECK_OK(EvaluateModelForShard(shard_id, image_label, model_labels, in EvaluateModel()
Dimagenet_accuracy_eval.cc59 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats,
68 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats, in OnSingleImageEvaluationComplete() argument
82 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats,
104 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats, in OnSingleImageEvaluationComplete() argument
Dimagenet_model_evaluator.h93 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats,
/external/tensorflow/tensorflow/core/protobuf/
Dtensor_bundle.proto48 // File "shard_id": bytes [offset, offset + size).
49 int32 shard_id = 3; field
60 // "shard_id", "offset", "size", "crc32c": all IGNORED.
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc154 int shard_id; in RegisterPerTableLoadOpsForAlgorithmBody() local
155 TF_RETURN_IF_ERROR(c->GetAttr("shard_id", &shard_id)); in RegisterPerTableLoadOpsForAlgorithmBody()
271 int shard_id; in RegisterPerTableRetrieveOpsForAlgorithmBody() local
272 TF_RETURN_IF_ERROR(c->GetAttr("shard_id", &shard_id)); in RegisterPerTableRetrieveOpsForAlgorithmBody()
/external/autotest/database/
Dschema_129.sql299 `shard_id` int(11) DEFAULT NULL,
304 KEY `hosts_to_shard_ibfk` (`shard_id`),
306 CONSTRAINT `hosts_to_shard_ibfk` FOREIGN KEY (`shard_id`) REFERENCES `afe_shards` (`id`)
413 `shard_id` int(11) DEFAULT NULL,
421 KEY `jobs_to_shard_ibfk` (`shard_id`),
425 CONSTRAINT `jobs_to_shard_ibfk` FOREIGN KEY (`shard_id`) REFERENCES `afe_shards` (`id`)
659 `shard_id` int(11) NOT NULL,
663 KEY `shard_shard_id_fk` (`shard_id`),
665 CONSTRAINT `shard_shard_id_fk` FOREIGN KEY (`shard_id`) REFERENCES `afe_shards` (`id`)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py843 shard_id=host_id))
861 shard_id=host_id))
933 shard_id=host_id))
953 shard_id=host_id))
995 shard_id=host_id))
1016 shard_id=host_id))
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_support.py683 for shard_id in range(self._tpu_assignment.num_towers):
687 with ops.device('/device:TPU:%d' % shard_id):
694 name='infeed-enqueue-%s-%d' % (spec.name, shard_id)))
700 name='infeed-enqueue-%s-%d' % (execution_mode, shard_id),
701 device_ordinal=shard_id))
838 for shard_id in range(self._tpu_assignment.num_towers):
843 shard_infeed_tensors[shard_id],
845 name='infeed-enqueue-%s-%d' % (execution_mode, shard_id),
846 device_ordinal=shard_id))
1104 for shard_id in range(self._tpu_assignment.num_towers):
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops_3d.cc531 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
532 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute()
533 T* input_data = input_backprop_data + shard_id * input_offset; in Compute()
534 const T* out_data = out_backprop_data + shard_id * output_offset; in Compute()
951 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
952 const T* input_data_shard = input_data + shard_id * input_offset; in Compute()
953 T* col_data_shard = col_buffer_data + shard_id * size_A; in Compute()
Dconv_grad_input_ops.cc508 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
509 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute()
510 T* input_data = input_backprop_data + shard_id * input_offset; in Compute()
511 const T* out_data = out_backprop_data + shard_id * output_offset; in Compute()
Dconv_grad_filter_ops.cc371 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
372 const T* input_data_shard = input_data + shard_id * input_offset; in Compute()
373 T* col_data_shard = col_buffer_data + shard_id * size_A; in Compute()
/external/autotest/scheduler/
Drdb_hosts.py116 self.shard_id = host.shard_id
Dscheduler_models_unittest.py313 self.assertIsNone(hqe.job.shard_id)
316 self.assertEquals(hqe.job.shard_id, 3)
Dmonitor_db.py831 if host.shard_id is not None and not server_utils.is_shard():
1074 if entry.job.shard_id is not None and not server_utils.is_shard():
1085 entry.job.shard_id, entry)
Drdb.py466 [host.shard_id for host in hosts])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1697 …argspec: "args=[\'parameters\', \'momenta\', \'velocities\', \'num_shards\', \'shard_id\', \'table…
1701 …momenta\', \'velocities\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1705 …argspec: "args=[\'parameters\', \'accumulators\', \'updates\', \'num_shards\', \'shard_id\', \'tab…
1709 …cumulators\', \'updates\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1713 …argspec: "args=[\'parameters\', \'accumulators\', \'num_shards\', \'shard_id\', \'table_id\', \'ta…
1717 …eters\', \'accumulators\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1721 …argspec: "args=[\'parameters\', \'ms\', \'mom\', \'mg\', \'num_shards\', \'shard_id\', \'table_id\…
1725 …argspec: "args=[\'parameters\', \'accumulators\', \'linears\', \'num_shards\', \'shard_id\', \'tab…
1729 …cumulators\', \'linears\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1733 …eters\', \'accumulators\', \'weights\', \'benefits\', \'num_shards\', \'shard_id\', \'table_id\', …
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1697 …argspec: "args=[\'parameters\', \'momenta\', \'velocities\', \'num_shards\', \'shard_id\', \'table…
1701 …momenta\', \'velocities\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1705 …argspec: "args=[\'parameters\', \'accumulators\', \'updates\', \'num_shards\', \'shard_id\', \'tab…
1709 …cumulators\', \'updates\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1713 …argspec: "args=[\'parameters\', \'accumulators\', \'num_shards\', \'shard_id\', \'table_id\', \'ta…
1717 …eters\', \'accumulators\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1721 …argspec: "args=[\'parameters\', \'ms\', \'mom\', \'mg\', \'num_shards\', \'shard_id\', \'table_id\…
1725 …argspec: "args=[\'parameters\', \'accumulators\', \'linears\', \'num_shards\', \'shard_id\', \'tab…
1729 …cumulators\', \'linears\', \'gradient_accumulators\', \'num_shards\', \'shard_id\', \'table_id\', …
1733 …eters\', \'accumulators\', \'weights\', \'benefits\', \'num_shards\', \'shard_id\', \'table_id\', …
[all …]
/external/autotest/venv/lucifer/
Dhandlers.py193 if self._job.shard_id is not None:
195 self._job.shard_id = None
/external/tensorflow/tensorflow/go/op/
Dwrappers.go9640 …elocities tf.Output, gradient_accumulators tf.Output, num_shards int64, shard_id int64, optional .…
9644 attrs := map[string]interface{}{"num_shards": num_shards, "shard_id": shard_id}
9794 …DescentParameters(scope *Scope, parameters tf.Output, num_shards int64, shard_id int64, optional .…
9798 attrs := map[string]interface{}{"num_shards": num_shards, "shard_id": shard_id}
10649 func RetrieveTPUEmbeddingADAMParameters(scope *Scope, num_shards int64, shard_id int64, optional ..…
10653 attrs := map[string]interface{}{"num_shards": num_shards, "shard_id": shard_id}
14766 …umulators tf.Output, gradient_accumulators tf.Output, num_shards int64, shard_id int64, optional .…
14770 attrs := map[string]interface{}{"num_shards": num_shards, "shard_id": shard_id}
14820 …*Scope, parameters tf.Output, accumulators tf.Output, num_shards int64, shard_id int64, optional .…
14824 attrs := map[string]interface{}{"num_shards": num_shards, "shard_id": shard_id}
[all …]
/external/autotest/frontend/afe/
Drpc_interface_unittest.py1806 shard_id = rpc_interface.add_shard(
1814 shard = models.Shard.objects.get(pk=shard_id)
1830 shard_id = rpc_interface.add_board_to_shard(
1840 shard = models.Shard.objects.get(pk=shard_id)
Dmodels.py1473 if self.shard_id and self.shard_id != shard.id:
1476 'from shard %s.' % (self.id, self.shard_id, updated_serialized,

12