Searched refs:input_spec_ (Results 1 – 4 of 4) sorted by relevance
88 if (feature >= input_spec_.dense_features_size()) { in GetExampleValue()96 if (feature_id >= input_spec_.dense_features_size()) { in GetExampleValue()124 int32 num_total_features = input_spec_.dense_features_size(); in RandomSample()140 *type = input_spec_.GetDenseFeatureType(rand_feature); in RandomSample()143 sparse_input_start + rand_feature - input_spec_.dense_features_size(); in RandomSample()145 (*sparse_indices_)(sparse_index, 1) + input_spec_.dense_features_size(); in RandomSample()151 *type = input_spec_.sparse(0).original_type(); in RandomSample()
44 input_spec_(input_spec), in TensorDataSet()47 for (int i = 0; i < input_spec_.dense_size(); ++i) { in TensorDataSet()48 for (int j = 0; j < input_spec_.dense(i).size(); ++j) { in TensorDataSet()118 const tensorforest::TensorForestDataSpec input_spec_; variable
171 input_spec_.ParseFromString(serialized_proto); in TreePredictionsV4Op()181 std::unique_ptr<TensorDataSet> data_set(new TensorDataSet(input_spec_, 0)); in Compute()258 tensorforest::TensorForestDataSpec input_spec_; member in tensorflow::tensorforest::TreePredictionsV4Op273 input_spec_.ParseFromString(serialized_proto); in TraverseTreeV4Op()282 std::unique_ptr<TensorDataSet> data_set(new TensorDataSet(input_spec_, 0)); in Compute()319 tensorforest::TensorForestDataSpec input_spec_; member in tensorflow::tensorforest::TraverseTreeV4Op
237 input_spec_.ParseFromString(serialized_proto); in ProcessInputOp()250 new TensorDataSet(input_spec_, random_seed_)); in Compute()351 tensorforest::TensorForestDataSpec input_spec_; member in tensorflow::tensorforest::ProcessInputOp412 tensorforest::TensorForestDataSpec input_spec_; member in tensorflow::tensorforest::GrowTreeOp