Searched defs:reduction_indices (Results 1 - 7 of 7) sorted by relevance
/external/eigen/unsupported/test/ |
H A D | cxx11_tensor_index_list.cpp | 319 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 D | reduce_join_op.cc | 76 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 D | quantized_instance_norm.cc | 305 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 D | math_grad.cc | 671 // 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 D | layout_optimizer_test.cc | 700 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 D | constant_folding.cc | 518 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 D | common_shape_fns.cc | 1042 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 671 milliseconds