Home
last modified time | relevance | path

Searched refs:vocab_file (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcandidate_sampler_ops.cc156 string vocab_file; in FixedUnigramCandidateSamplerOp() local
157 OP_REQUIRES_OK(context, context->GetAttr("vocab_file", &vocab_file)); in FixedUnigramCandidateSamplerOp()
161 context, !vocab_file.empty() || !unigrams.empty(), in FixedUnigramCandidateSamplerOp()
163 OP_REQUIRES(context, vocab_file.empty() || unigrams.empty(), in FixedUnigramCandidateSamplerOp()
176 if (!vocab_file.empty()) { in FixedUnigramCandidateSamplerOp()
177 set_sampler(new FixedUnigramSampler(context->env(), range_max, vocab_file, in FixedUnigramCandidateSamplerOp()
Drange_sampler.cc233 const string& vocab_file, in FixedUnigramSampler() argument
243 TF_CHECK_OK(LoadFromFile(env, vocab_file, distortion)); in FixedUnigramSampler()
281 Status FixedUnigramSampler::LoadFromFile(Env* env, const string& vocab_file, in LoadFromFile() argument
284 TF_RETURN_IF_ERROR(env->NewRandomAccessFile(vocab_file, &file)); in LoadFromFile()
Drange_sampler.h215 FixedUnigramSampler(Env* env, int64 range, const string& vocab_file,
245 Status LoadFromFile(Env* env, const string& vocab_file, float distortion);
Dlookup_util.cc31 Status GetNumLinesInTextFile(Env* env, const string& vocab_file, in GetNumLinesInTextFile() argument
34 TF_RETURN_IF_ERROR(env->NewRandomAccessFile(vocab_file, &file)); in GetNumLinesInTextFile()
/external/tensorflow/tensorflow/contrib/lookup/
Dlookup_ops_test.py1049 vocab_file = self._createVocabFile("feat_to_id_1.txt")
1055 vocab_file, vocab_size=vocab_size),
1067 vocab_file = self._createVocabFile("feat_to_id_2.txt")
1073 vocab_file, vocab_size=vocab_size),
1084 vocab_file = self._createVocabFile(
1091 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
1110 vocab_file = self._createVocabFile("feat_to_id_1.txt")
1118 vocab_file, vocab_size=vocab_size),
1131 vocab_file = self._createVocabFile("feat_to_id_2.txt", ("42", "1", "-1000"))
1139 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
[all …]
/external/tensorflow/tensorflow/python/ops/
Dcandidate_sampling_ops.py221 vocab_file='', argument
302 vocab_file=vocab_file, distortion=distortion,
/external/tensorflow/tensorflow/python/kernel_tests/
Dlookup_ops_test.py667 vocab_file = self._createVocabFile("feat_to_id_1.txt")
672 vocab_file, vocab_size=vocab_size)
686 vocab_file = self._createVocabFile("feat_to_id_2.txt")
691 vocab_file, vocab_size=vocab_size)
703 vocab_file = self._createVocabFile(
709 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64)
729 vocab_file = self._createVocabFile("feat_to_id_1.txt")
733 vocab_file, vocab_size=vocab_size), oov_buckets)
744 vocab_file = self._createVocabFile("feat_to_id_2.txt", ("42", "1", "-1000"))
749 vocab_file, vocab_size=vocab_size, key_dtype=dtypes.int64),
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_FixedUnigramCandidateSampler.pbtxt61 name: "vocab_file"
66 corresponding to the count or relative probability. Exactly one of vocab_file
110 order. Exactly one of vocab_file and unigrams should be passed to this op.
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.random.pbtxt17 …classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'vocab_file\', \'distortion\…
Dtensorflow.nn.pbtxt161 …classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'vocab_file\', \'distortion\…
Dtensorflow.raw_ops.pbtxt1305 …classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'vocab_file\', \'distortion\…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.random.pbtxt17 …classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'vocab_file\', \'distortion\…
Dtensorflow.nn.pbtxt201 …classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'vocab_file\', \'distortion\…
Dtensorflow.raw_ops.pbtxt1305 …classes\', \'num_true\', \'num_sampled\', \'unique\', \'range_max\', \'vocab_file\', \'distortion\…
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column_test.py749 vocab_file = os.path.join(self.get_temp_dir(), file_name)
750 with open(vocab_file, 'w') as f:
752 return vocab_file
/external/tensorflow/tensorflow/python/feature_column/
Dsequence_feature_column_test.py797 vocab_file = os.path.join(self.get_temp_dir(), file_name)
798 with open(vocab_file, 'w') as f:
800 return vocab_file
/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util_test.py46 vocab_file = os.path.join(self.get_temp_dir(), file_name)
47 with open(vocab_file, "w") as f:
49 return vocab_file
/external/tensorflow/tensorflow/contrib/eager/python/examples/workshop/
D3_inspecting.ipynb138 "vocab_file = os.path.join(gs_data_dir, vocab_name)\n",
139 "!gsutil cp {vocab_file} {data_dir}\n",
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator_test.py217 vocab_file = gfile.GFile(vocab_file_name, mode='w')
218 vocab_file.write(VOCAB_FILE_CONTENT)
219 vocab_file.close()
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt12353 name: "vocab_file"
Dops_history.v2.pbtxt26565 name: "vocab_file"
26666 name: "vocab_file"
Dops_history.v1.pbtxt26936 name: "vocab_file"
27037 name: "vocab_file"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt13256 name: "vocab_file"