OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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