Searched refs:reduction_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_pooling.h | 112 Eigen::array<int, 2> reduction_dims; variable
114 reduction_dims[0] = 1;
115 reduction_dims[1] = 2;
117 reduction_dims[0] = 2;
118 reduction_dims[1] = 3;
127 Eigen::type2index<3> > >::type reduction_dims; variable
136 .maximum(reduction_dims)
242 Eigen::array<int, 1> reduction_dims; in CuboidMaxPooling() local
243 reduction_dims[0] = 1; in CuboidMaxPooling()
247 Eigen::IndexList<Eigen::type2index<1> > reduction_dims; in CuboidMaxPooling() local
[all …]
|
| D | reduce_join_op.cc | 79 const int32 reduction_dims = reduction_indices_flat.size(); in GetReducedIndices() local
81 gtl::InlinedVector<int32, 8> reduced_indices(reduction_dims); in GetReducedIndices()
82 for (int32 i = 0; i < reduction_dims; ++i) { in GetReducedIndices()
83 reduced_indices[i] = reduction_indices_flat(reduction_dims - i - 1); in GetReducedIndices()
131 const int32 reduction_dims = reduction_indices_flat.size(); in Compute() local
134 for (int32 i = 0; i < reduction_dims; i++) { in Compute()
|
| D | tile_ops.cc | 360 std::vector<Tmultiples> reduction_dims; in HandleCaseImpl() local
368 reduction_dims.push_back(i); in HandleCaseImpl()
375 if (reduction_dims.size() == (D)) { \ in HandleCaseImpl()
376 HandleReduce<T, NDIM, (D)>(context, reduction_dims, result); \ in HandleCaseImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 174 std::vector<int64> reduction_dims(input_dims - 1); in Compile() local
175 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index, in Compile()
177 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(), in Compile()
189 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
202 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 752 NodeDef reduction_dims; in QuantizeNodes() local
753 reduction_dims.set_op("Const"); in QuantizeNodes()
754 reduction_dims.set_name(unique_input_name + "/reduction_dims"); in QuantizeNodes()
755 AddNodeInput("^" + NodeNameFromInput(input_name), &reduction_dims); in QuantizeNodes()
756 SetNodeAttr("dtype", DT_INT32, &reduction_dims); in QuantizeNodes()
760 &reduction_dims); in QuantizeNodes()
761 new_nodes->push_back(reduction_dims); in QuantizeNodes()
777 AddNodeInput(reduction_dims.name(), &min_node); in QuantizeNodes()
786 AddNodeInput(reduction_dims.name(), &max_node); in QuantizeNodes()
|