Searched refs:reduce_dim (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduce_join_op_test.py | 48 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/ |
| D | tile_ops_impl.h | 62 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
64 out.device(d) = in.sum(reduce_dim).reshape(reshape_dim);
|
| D | reduce_join_op.cc | 68 for (int32 reduce_dim : reduced_indices) { in GetReductionIterSize() local
69 result *= input_shape.dim_size(reduce_dim); in GetReductionIterSize()
|
| D | bincount_op.cc | 76 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
78 partial_bins.any(reduce_dim).cast<T>(); in Compute()
127 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
128 output.device(context->eigen_cpu_device()) = partial_bins.sum(reduce_dim); in Compute()
|
| D | tile_ops.cc | 75 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
492 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
496 reduce_dim[i] = reduce_dim_in[i]; in HandleReduce()
505 context->input(0).tensor<T, NDIM>(), reduce_dim, reshape_dim); in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 92 int64 reduce_dim = n_dims - 1; in GetMatrixDiagonal() local
94 reduce_dim = n_dims - 2; in GetMatrixDiagonal()
99 {reduce_dim}); in GetMatrixDiagonal()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 3196 reduce_dim = _ReductionDims(input_tensor, axis)
3198 input_tensor, axis=axis, keepdims=True, dims=reduce_dim)
3208 dims=reduce_dim))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1048 int reduce_dim = in TEST_F() local
1053 .Input(test::AsScalar<int32>(reduce_dim)) in TEST_F()
1064 int reduce_dim = in TEST_F() local
1069 .Input(test::AsScalar<int32>(reduce_dim)) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2715 auto reduce_dim = rewriter.create<TF::ConstOp>( in matchAndRewrite() local
2725 rewriter.create<TF::MaxOp>(loc, logits, reduce_dim, in matchAndRewrite()
2737 rewriter.create<TF::SumOp>(loc, exp, reduce_dim, in matchAndRewrite()
|