HomeSort by relevance Sort by last modified time
    Searched refs:reduce_dim (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/contrib/boosted_trees/ops/
prediction_ops.cc 33 bool reduce_dim; local
34 c->GetAttr("reduce_dim", &reduce_dim).IgnoreError();
37 reduce_dim ? learner_config.num_classes() - 1
52 .Attr("reduce_dim: bool")
76 reduce_dim: whether to reduce the dimension (legacy impl) or not.
  /external/tensorflow/tensorflow/core/kernels/
tile_ops_impl.h 62 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
64 out.device(d) = in.sum(reduce_dim).reshape(reshape_dim);
reduce_join_op.cc 68 for (int32 reduce_dim : reduced_indices) {
69 result *= input_shape.dim_size(reduce_dim);
tile_ops.cc 77 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
418 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; local
422 reduce_dim[i] = reduce_dim_in[i];
431 context->input(0).tensor<T, NDIM>(), reduce_dim, reshape_dim);
  /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
prediction_ops_test.py 153 reduce_dim=False):
166 reduce_dim=reduce_dim)
186 reduce_dim=True)
213 reduce_dim=True)
243 reduce_dim=True)
287 reduce_dim=True)
376 reduce_dim=True)
424 reduce_dim=True)
471 reduce_dim=True
    [all...]
model_ops_test.py 129 reduce_dim=True)
190 reduce_dim=True)
256 reduce_dim=True)
285 reduce_dim=True)
308 reduce_dim=True)
  /external/tensorflow/tensorflow/python/kernel_tests/
reduce_join_op_test.py 47 def _joined_array(num_dims, reduce_dim):
52 reduce_dim: The dimension to reduce.
62 flat[i] = "".join([(dims[:reduce_dim] + "%d" + dims[reduce_dim:]) % j
89 output_array_dim_zero = _joined_array(num_dims, reduce_dim=0)
90 output_array_dim_one = _joined_array(num_dims, reduce_dim=1)
91 output_array_dim_two = _joined_array(num_dims, reduce_dim=2)
211 truths = [_joined_array(num_dims=5, reduce_dim=i) for i in xrange(5)]
219 truths = [_joined_array(num_dims=5, reduce_dim=i) for i in xrange(5)]
  /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
prediction_ops.cc 117 bool reduce_dim; local
118 OP_REQUIRES_OK(context, context->GetAttr("reduce_dim", &reduce_dim));
119 prediction_vector_size_ = reduce_dim ? num_classes_ - 1 : num_classes_;
  /external/tensorflow/tensorflow/compiler/tests/
randomized_tests.cc     [all...]
  /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
gbdt_batch.py 385 reduce_dim=self._reduce_dim)
    [all...]

Completed in 166 milliseconds