Searched refs:reduction_dims (Results 1 – 7 of 7) 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 | 427 std::vector<Tmultiples> reduction_dims; in HandleCaseImpl() local
435 reduction_dims.push_back(i); in HandleCaseImpl()
442 if (reduction_dims.size() == (D)) { \ in HandleCaseImpl()
443 HandleReduce<T, NDIM, (D)>(context, reduction_dims, result); \ in HandleCaseImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 300 std::vector<int64> reduction_dims(input_dims - 1); in Compile() local
301 std::iota(reduction_dims.begin(), reduction_dims.begin() + feature_index, in Compile()
303 std::iota(reduction_dims.begin() + feature_index, reduction_dims.end(), in Compile()
315 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
328 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 757 NodeDef reduction_dims; in QuantizeNodes() local
758 reduction_dims.set_op("Const"); in QuantizeNodes()
759 reduction_dims.set_name(unique_input_name + "/reduction_dims"); in QuantizeNodes()
760 AddNodeInput("^" + NodeNameFromInput(input_name), &reduction_dims); in QuantizeNodes()
761 SetNodeAttr("dtype", DT_INT32, &reduction_dims); in QuantizeNodes()
765 &reduction_dims); in QuantizeNodes()
766 new_nodes->push_back(reduction_dims); in QuantizeNodes()
782 AddNodeInput(reduction_dims.name(), &min_node); in QuantizeNodes()
791 AddNodeInput(reduction_dims.name(), &max_node); in QuantizeNodes()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 130 ArrayRef<int64_t> reduction_dims) { in GetTransposeMapForReduction() argument
132 for (auto dim : reduction_dims) s.insert(dim); in GetTransposeMapForReduction()
137 for (auto dim : reduction_dims) permutation.push_back(dim); in GetTransposeMapForReduction()
940 SmallVector<int, 4> reduction_dims; in matchAndRewrite() local
942 reduction_dims.push_back(dim.getSExtValue()); in matchAndRewrite()
949 bool is_reduced = llvm::is_contained(reduction_dims, i); in matchAndRewrite()
1273 ArrayRef<int64_t> reduction_dims) { in GetReduceOpInitTensorDynSizes() argument
1275 for (auto dim : reduction_dims) s.insert(dim); in GetReduceOpInitTensorDynSizes()
1333 SmallVector<int64_t, 4> reduction_dims = Extract1DVector(op.dimensions()); in matchAndRewrite() local
1335 rewriter.getContext(), src_rank, reduction_dims)); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_op_interfaces.td | 80 Example: Mean(Transpose($arg, $reduction_dims))
81 -> Mean($arg, Transpose($reduction_dims))
|