Home
last modified time | relevance | path

Searched refs:reduce_dim (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dprediction_ops.cc33 bool reduce_dim; in ApplyGradientTreesPredictionShapeFn() local
34 c->GetAttr("reduce_dim", &reduce_dim).IgnoreError(); in ApplyGradientTreesPredictionShapeFn()
37 reduce_dim ? learner_config.num_classes() - 1 in ApplyGradientTreesPredictionShapeFn()
49 bool reduce_dim; in ApplyGradientTreesPredictionVerboseShapeFn() local
50 c->GetAttr("reduce_dim", &reduce_dim).IgnoreError(); in ApplyGradientTreesPredictionVerboseShapeFn()
53 reduce_dim ? learner_config.num_classes() - 1 in ApplyGradientTreesPredictionVerboseShapeFn()
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dprediction_ops_test.py169 reduce_dim=False): argument
182 reduce_dim=reduce_dim)
190 reduce_dim=False): argument
201 reduce_dim=reduce_dim)
221 reduce_dim=True)
248 reduce_dim=True)
278 reduce_dim=True)
322 reduce_dim=True)
368 reduce_dim=True)
458 reduce_dim=True)
[all …]
Dmodel_ops_test.py129 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/
Dreduce_join_op_test.py48 def _joined_array(num_dims, reduce_dim): argument
63 flat[i] = "".join([(dims[:reduce_dim] + "%d" + dims[reduce_dim:]) % j
90 output_array_dim_zero = _joined_array(num_dims, reduce_dim=0)
91 output_array_dim_one = _joined_array(num_dims, reduce_dim=1)
92 output_array_dim_two = _joined_array(num_dims, reduce_dim=2)
191 truths = [_joined_array(num_dims=5, reduce_dim=i) for i in xrange(5)]
198 truths = [_joined_array(num_dims=5, reduce_dim=i) for i in xrange(5)]
/external/tensorflow/tensorflow/core/kernels/
Dtile_ops_impl.h62 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
64 out.device(d) = in.sum(reduce_dim).reshape(reshape_dim);
Dreduce_join_op.cc68 for (int32 reduce_dim : reduced_indices) { in GetReductionIterSize() local
69 result *= input_shape.dim_size(reduce_dim); in GetReductionIterSize()
Dtile_ops.cc78 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
425 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
429 reduce_dim[i] = reduce_dim_in[i]; in HandleReduce()
438 context->input(0).tensor<T, NDIM>(), reduce_dim, reshape_dim); in HandleReduce()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc118 bool reduce_dim; in GradientTreesPredictionOp() local
119 OP_REQUIRES_OK(context, context->GetAttr("reduce_dim", &reduce_dim)); in GradientTreesPredictionOp()
120 prediction_vector_size_ = reduce_dim ? num_classes_ - 1 : num_classes_; in GradientTreesPredictionOp()
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py508 reduce_dim=self._reduce_dim)
526 reduce_dim=self._reduce_dim)
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1047 int reduce_dim = in TEST_F() local
1052 .Input(test::AsScalar<int32>(reduce_dim)) in TEST_F()
1063 int reduce_dim = in TEST_F() local
1068 .Input(test::AsScalar<int32>(reduce_dim)) in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc1329 const int64 reduce_dim = rhs_kept_dim < rhs_collapsing_dim in HandleDotStrengthReduction() local
1334 multiply(new_lhs, new_rhs), reduce_dim)))); in HandleDotStrengthReduction()