Searched defs:reduction_indices (Results 1 - 7 of 7) sorted by relevance

/external/eigen/unsupported/test/
H A Dcxx11_tensor_index_list.cpp319 ReductionIndices reduction_indices; local
320 reduction_indices.set(1, 1);
321 reduction_indices.set(3, 3);
322 EIGEN_STATIC_ASSERT((internal::array_get<0>(reduction_indices) == 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
323 EIGEN_STATIC_ASSERT((internal::array_get<2>(reduction_indices) == 2), YOU_MADE_A_PROGRAMMING_MISTAKE);
345 Tensor<float, 0> result2 = tensor.sum(reduction_indices);
/external/tensorflow/tensorflow/core/kernels/
H A Dreduce_join_op.cc76 gtl::InlinedVector<int32, 8> GetReducedIndices(const Tensor& reduction_indices, argument
78 const auto reduction_indices_flat = reduction_indices.flat<int32>();
129 const Tensor& reduction_indices = context->input(1); local
130 const auto reduction_indices_flat = reduction_indices.flat<int32>();
149 GetReducedIndices(reduction_indices, input_dims);
H A Dquantized_instance_norm.cc305 reduction_indices; variable
315 const Eigen::array<Index, 2> reduction_indices{1, 2};
351 reduction_indices, Eigen::internal::MeanReducer<float>());
358 .reduce(reduction_indices, Eigen::internal::MeanReducer<float>());
/external/tensorflow/tensorflow/cc/gradients/
H A Dmath_grad.cc671 // reduction_indices = [1]
704 // Stop propagation along reduction_indices
731 // Stop propagation along reduction_indices
784 // reduction_indices: [1]
786 auto reduction_indices = op.input(1); local
793 ReducedShapeHelper(scope, input_shape, reduction_indices);
814 auto num_selected = Reshape(scope, Sum(scope, indicators, reduction_indices),
825 // Stop propagation along reduction_indices
876 auto reduction_indices = Reshape(scope, op.input(1), {-1}); local
880 ReducedShapeHelper(scope, input_shape, reduction_indices);
[all...]
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dlayout_optimizer_test.cc700 auto reduction_indices = local
701 ops::Const(s.WithOpName("reduction_indices"), {0, 1, 2}, {3});
702 auto sum = ops::Sum(s.WithOpName("sum"), conv, reduction_indices);
715 EXPECT_EQ(sum_node->input(1), "LayoutOptimizer-sum-reduction_indices");
716 auto sum_const = node_map.GetNode("LayoutOptimizer-sum-reduction_indices");
H A Dconstant_folding.cc518 int reduction_indices = reduce_dims[j].size(); local
519 Tensor value(type, TensorShape({reduction_indices}));
520 for (int i = 0; i < reduction_indices; ++i) {
627 string const_name = OptimizedNodeName(*node, "-reduction_indices");
631 NodeDef* reduction_indices = graph_->add_node(); local
640 *reduction_indices = CreateNodeDef(const_name, TensorValue(&value));
641 reduction_indices->set_device(node->device());
644 *reduction_indices->add_input() = ctrl_dep;
645 node_map_->AddNode(const_name, reduction_indices);
648 node->set_input(1, reduction_indices
[all...]
/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc1042 auto reduction_indices = reduction_indices_t->flat<T>(); local
1044 const T reduction_index = reduction_indices(i);
1100 "reduction_indices can only be int32 or int64");

Completed in 816 milliseconds