| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | dropout_utils_test.cc | 46 weights_.push_back(1.1 + 0.4 * i); in SetUp()
51 std::vector<float> weights_; member in tensorflow::boosted_trees::utils::__anon8abe69950111::DropoutUtilsTest
66 weights_, &dropped_trees, in TEST_F()
80 weights_, &dropped_trees, in TEST_F()
86 EXPECT_EQ(original_weights, weights_); in TEST_F()
102 weights_, &dropped_trees, in TEST_F()
133 weights_, &dropped_trees, in TEST_F()
139 EXPECT_EQ(original_weights, weights_); in TEST_F()
154 weights_, &dropped_trees, in TEST_F()
172 weights_, &dropped_trees, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | mfcc_mel_filterbank.cc | 114 weights_.resize(input_length_); in Initialize()
118 weights_[i] = 0.0; in Initialize()
121 weights_[i] = in Initialize()
125 weights_[i] = (center_frequencies_[0] - FreqToMel(i * hz_per_sbin)) / in Initialize()
139 band_weights_sum += (1.0 - weights_[i]); in Initialize()
141 band_weights_sum += weights_[i]; in Initialize()
189 double weighted = spec_val * weights_[i]; in Compute()
|
| D | mfcc_mel_filterbank.h | 52 std::vector<double> weights_; // Right-side weight for this fft bin. variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mfcc_mel_filterbank.cc | 115 weights_.resize(input_length_); in Initialize()
119 weights_[i] = 0.0; in Initialize()
122 weights_[i] = in Initialize()
126 weights_[i] = (center_frequencies_[0] - FreqToMel(i * hz_per_sbin)) / in Initialize()
140 band_weights_sum += (1.0 - weights_[i]); in Initialize()
142 band_weights_sum += weights_[i]; in Initialize()
188 double weighted = spec_val * weights_[i]; in Compute()
|
| D | range_sampler.cc | 244 CHECK_EQ(range, weights_.size()); in FixedUnigramSampler()
245 dist_sampler_.reset(new random::DistributionSampler(weights_)); in FixedUnigramSampler()
260 CHECK_EQ(range, weights_.size()); in FixedUnigramSampler()
261 dist_sampler_.reset(new random::DistributionSampler(weights_)); in FixedUnigramSampler()
265 if (value < 0 || static_cast<size_t>(value) >= weights_.size()) { in Probability()
268 return weights_.at(value) / total_weight_; in Probability()
277 if (word_id % num_shards_ == shard_) weights_.push_back(0.0); in FillReservedIds()
288 int32 word_id = weights_.size(); in LoadFromFile()
303 weights_.push_back(w); in LoadFromFile()
312 int32 word_id = weights_.size(); in LoadFromUnigrams()
[all …]
|
| D | mfcc_mel_filterbank.h | 52 std::vector<double> weights_; // Right-side weight for this fft bin. variable
|
| D | range_sampler.h | 232 std::vector<float> weights_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected_test.cc | 151 weights_ =
162 auto bias_scale = GetScale(input_) * GetScale(weights_);
179 {GetShape(input_), GetShape(weights_),
189 int weights_; member in tflite::__anon3e7c6b6f0111::BaseFullyConnectedOpModel
204 void SetWeights(const std::vector<float>& f) { PopulateTensor(weights_, f); } in SetWeights()
225 QuantizeAndPopulate<T>(weights_, data); in SetWeights()
245 TfLiteTensor* t = interpreter_->tensor(weights_); in ShuffleAndSetWeights()
251 PopulateTensor(weights_, 0, quantized_data.data(), in ShuffleAndSetWeights()
287 weights_ = AddInput(weights);
301 BuildInterpreter({GetShape(input_), GetShape(weights_), GetShape(bias_)});
[all …]
|
| D | basic_rnn_test.cc | 181 weights_ = AddInput(weights); in RNNOpModel()
199 PopulateTensor(weights_, f); in SetWeights()
222 int weights_; member in tflite::__anon35c533a00111::RNNOpModel
251 void SetWeights(std::initializer_list<float> f) { SetWeights(weights_, f); } in SetWeights()
|
| D | embedding_lookup_sparse_test.cc | 42 weights_ = AddInput(TensorType_FLOAT32); in EmbeddingLookupSparseOpModel()
59 PopulateTensor(weights_, weights_data); in SetInput()
80 int weights_; member in tflite::__anon2b9276160111::EmbeddingLookupSparseOpModel
|
| D | unidirectional_sequence_rnn_test.cc | 183 weights_ = AddInput(weights); in UnidirectionalRNNOpModel()
211 PopulateTensor(weights_, f); in SetWeights()
235 int weights_; member in tflite::__anon2f8e60700111::UnidirectionalRNNOpModel
266 void SetWeights(std::initializer_list<float> f) { SetWeights(weights_, f); } in SetWeights()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 288 return weights_; in weights()
293 return weights_; in weights()
337 TRT_ShapedWeights weights_; variable
|
| D | convert_nodes.cc | 716 : weights_(weights), initialized_(true), is_tensor_(false) {} in TRT_TensorOrWeights()
722 weights_(rhs.weights_), in TRT_TensorOrWeights()
730 weights_ = rhs.weights_; in operator =()
759 StrAppend(&output, "weights=", weights_.DebugString()); in DebugString()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 533 weights_ = AddInput(weights); in FullyConnectedOpModel()
542 auto bias_scale = GetScale(input_) * GetScale(weights_); in FullyConnectedOpModel()
552 BuildInterpreter({GetShape(input_), GetShape(weights_), GetShape(bias_)}); in FullyConnectedOpModel()
560 SetData(weights_, weights_type_, data); in SetWeights()
573 int weights_; member in tflite::__anon6d004b400111::FullyConnectedOpModel
1908 weights_ = AddInput(weights); in RNNOpModel()
1926 PopulateTensor(weights_, f); in SetWeights()
1949 int weights_; member in tflite::__anon6d004b400111::RNNOpModel
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 972 weights_(std::move(weights)), in CudnnRnnParamsDescriptor()
986 return weights_; in params_weights()
995 ParamsRegions weights_; member in stream_executor::gpu::__anonf41f84630311::CudnnRnnParamsDescriptor
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1661 return weights_; in params_weights()
1673 ParamsRegions weights_; member in stream_executor::gpu::MIOpenRnnParamsDescriptor
|