| /external/tensorflow/tensorflow/python/ops/ |
| D | candidate_sampling_ops.py | 36 def uniform_candidate_sampler(true_classes, num_true, num_sampled, unique, argument
87 true_classes, num_true, num_sampled, unique, range_max, seed=seed1,
99 def log_uniform_candidate_sampler(true_classes, num_true, num_sampled, unique, argument
153 true_classes, num_true, num_sampled, unique, range_max, seed=seed1,
162 def learned_unigram_candidate_sampler(true_classes, num_true, num_sampled, argument
214 true_classes, num_true, num_sampled, unique, range_max, seed=seed1,
222 num_true, argument
306 true_classes, num_true, num_sampled, unique, range_max,
313 def all_candidate_sampler(true_classes, num_true, num_sampled, unique, argument
344 true_classes, num_true, num_sampled, unique, seed=seed1, seed2=seed2,
[all …]
|
| D | nn_test.py | 552 def _GenerateTestData(self, num_classes, dim, batch_size, num_true, labels, argument
595 (batch_size, num_true, dim)),
597 true_b = true_b.reshape((batch_size, num_true))
606 exp_labels = np.hstack((np.ones_like(true_logits) / num_true,
641 for num_true in range(1, 5):
643 low=0, high=num_classes, size=batch_size * num_true)
649 num_true=num_true,
657 labels, dtype=dtypes.int64, shape=(batch_size, num_true)),
661 num_true=num_true,
666 name="sampled_logits_basic_num_true_%d" % num_true)
[all …]
|
| D | nn_impl.py | 1815 num_true=1, argument
1887 num_true=num_true,
1938 new_true_w_shape = array_ops.concat([[-1, num_true], dim], 0)
1946 true_logits = array_ops.reshape(_sum_rows(dots_as_matrix), [-1, num_true])
1947 true_b = array_ops.reshape(true_b, [-1, num_true])
1953 labels, sampled, num_true=num_true)
1987 array_ops.ones_like(true_logits) / num_true,
2002 num_true=1, argument
2092 num_true=num_true,
2107 num_true=1, argument
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | candidate_sampling_ops.cc | 30 int64 num_true; in CandidateSamplerShapeFn() local
31 TF_RETURN_IF_ERROR(c->GetAttr("num_true", &num_true)); in CandidateSamplerShapeFn()
39 c->set_output(1, c->Matrix(batch_size, num_true)); in CandidateSamplerShapeFn()
145 int64 num_true; in __anon3d65770a0202() local
146 TF_RETURN_IF_ERROR(c->GetAttr("num_true", &num_true)); in __anon3d65770a0202()
153 c->WithValue(c->Dim(true_classes, 1), num_true, &unused)); in __anon3d65770a0202()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op.cc | 78 TTypes<int64>::UnalignedScalar num_true) { in Compute()
79 num_true() = CountAccumulator<T>(input.data(), input.data() + input.size()); in Compute()
143 int64 num_true; in Compute() local
144 TTypes<int64>::UnalignedScalar num_true_t(&num_true); in Compute()
150 TensorShape output_shape({num_true, input_dims}); in Compute()
214 TTypes<Tindex>::UnalignedScalar num_true); \
280 Tensor num_true; in ComputeAsyncType() local
283 TensorShape({}), &num_true), in ComputeAsyncType()
286 num_true.scalar<Tindex>().data()); in ComputeAsyncType()
314 Tindex num_true = *num_true_host.data(); in ComputeAsyncType() local
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AllCandidateSampler.pbtxt | 6 A batch_size * num_true matrix, in which each row contains the
7 IDs of the num_true target_classes in the corresponding original label.
20 A batch_size * num_true matrix, representing
35 name: "num_true"
|
| D | api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt | 7 A batch_size * num_true matrix, in which each row contains the
8 IDs of the num_true target_classes in the corresponding original label.
21 A batch_size * num_true matrix, representing
36 name: "num_true"
|
| D | api_def_LogUniformCandidateSampler.pbtxt | 6 A batch_size * num_true matrix, in which each row contains the
7 IDs of the num_true target_classes in the corresponding original label.
20 A batch_size * num_true matrix, representing
35 name: "num_true"
|
| D | api_def_LearnedUnigramCandidateSampler.pbtxt | 6 A batch_size * num_true matrix, in which each row contains the
7 IDs of the num_true target_classes in the corresponding original label.
20 A batch_size * num_true matrix, representing
35 name: "num_true"
|
| D | api_def_UniformCandidateSampler.pbtxt | 6 A batch_size * num_true matrix, in which each row contains the
7 IDs of the num_true target_classes in the corresponding original label.
20 A batch_size * num_true matrix, representing
35 name: "num_true"
|
| D | api_def_FixedUnigramCandidateSampler.pbtxt | 6 A batch_size * num_true matrix, in which each row contains the
7 IDs of the num_true target_classes in the corresponding original label.
20 A batch_size * num_true matrix, representing
35 name: "num_true"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.random.pbtxt | 17 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'seed\', \'name\'], …
29 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'voca…
41 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'seed…
45 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'seed…
109 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'seed…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.random.pbtxt | 17 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'seed\', \'name\'], …
29 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'voca…
45 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'seed…
49 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'seed…
121 …argspec: "args=[\'true_classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'seed…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | AllCandidateSampler.pbtxt | 20 name: "num_true"
69 name: "num_true"
|
| D | ThreadUnsafeUnigramCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | UniformCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | LogUniformCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | LearnedUnigramCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | FixedUnigramCandidateSampler.pbtxt | 20 name: "num_true"
121 name: "num_true"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | AllCandidateSampler.pbtxt | 20 name: "num_true"
69 name: "num_true"
|
| D | LogUniformCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | ThreadUnsafeUnigramCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | LearnedUnigramCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | UniformCandidateSampler.pbtxt | 20 name: "num_true"
75 name: "num_true"
|
| D | FixedUnigramCandidateSampler.pbtxt | 20 name: "num_true"
121 name: "num_true"
|