Home
last modified time | relevance | path

Searched refs:Axis (Results 1 – 25 of 179) sorted by relevance

12345678

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dshape.h32 enum class Axis { enum
44 std::string ToString(Axis t);
76 constexpr Axis GetAxis(int index);
79 Axis GetAxis(Layout layout, int32_t index);
83 constexpr int GetAxisIndex(Axis axis);
86 int GetAxisIndex(Layout layout, Axis axis);
90 constexpr bool HasAxis(Axis axis);
93 bool HasAxis(Layout layout, Axis axis);
114 template <Axis D>
116 int32_t get(Axis axis) const;
[all …]
Dshape_test.cc32 ASSERT_TRUE(OIHW.set<Axis::OUTPUT_CHANNELS>(2)); in TEST()
33 ASSERT_TRUE(OIHW.set(Axis::HEIGHT, 3)); in TEST()
38 ASSERT_FALSE(OIHW.set(Axis::CHANNELS, 10)); in TEST()
39 ASSERT_FALSE(OIHW.set<Axis::CHANNELS>(10)); in TEST()
42 EXPECT_EQ(1, OIHW.get(Axis::INPUT_CHANNELS)); in TEST()
45 EXPECT_EQ(4, OIHW.get<Axis::WIDTH>()); in TEST()
49 EXPECT_EQ(-1, OIHW.get(Axis::CHANNELS)); in TEST()
50 EXPECT_EQ(-1, OIHW.get<Axis::CHANNELS>()); in TEST()
54 std::vector<Axis> expected = {Axis::OUTPUT_CHANNELS, Axis::INPUT_CHANNELS, in TEST()
55 Axis::HEIGHT, Axis::WIDTH}; in TEST()
[all …]
Dshape.cc31 Axis operator()() const { in operator ()()
42 Axis axis;
54 std::string ToString(Axis axis) { in ToString()
56 case Axis::BATCH: in ToString()
58 case Axis::CHANNELS: in ToString()
60 case Axis::INPUT_CHANNELS: in ToString()
62 case Axis::OUTPUT_CHANNELS: in ToString()
64 case Axis::HEIGHT: in ToString()
66 case Axis::WIDTH: in ToString()
68 case Axis::VALUE: in ToString()
[all …]
Doperations.cc276 template <Axis T>
285 template <Axis T>
294 template <Axis T>
303 template <Axis T>
312 template <Axis T>
320 template <Axis T>
332 template <Axis AxisT, typename AttrT>
338 template <Axis AxisT, typename AttrT>
352 template <Axis AxisT>
362 template <Axis AxisT>
[all …]
Dmodel_builder_helper.cc123 Axis* axis) { in ExtractAxisFromIndex()
131 std::vector<Axis> index_to_axis; in ExtractAxisFromIndex()
135 index_to_axis = {Axis::BATCH}; in ExtractAxisFromIndex()
139 index_to_axis = {Axis::BATCH, Axis::CHANNELS}; in ExtractAxisFromIndex()
143 index_to_axis = {Axis::BATCH, Axis::WIDTH, Axis::CHANNELS}; in ExtractAxisFromIndex()
147 index_to_axis = {Axis::BATCH, Axis::HEIGHT, Axis::WIDTH, Axis::CHANNELS}; in ExtractAxisFromIndex()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpooling.cc42 std::map<Axis, std::string> axis_to_src_coord = { in GetAveragePoolingKernelCode()
43 {Axis::WIDTH, "x_c"}, {Axis::HEIGHT, "y_c"}, {Axis::DEPTH, "d_c"}, in GetAveragePoolingKernelCode()
44 {Axis::CHANNELS, "Z"}, {Axis::BATCH, "B"}, in GetAveragePoolingKernelCode()
47 std::map<Axis, std::string> axis_to_dst_coord = { in GetAveragePoolingKernelCode()
48 {Axis::WIDTH, "X"}, {Axis::HEIGHT, "Y"}, {Axis::DEPTH, "D"}, in GetAveragePoolingKernelCode()
49 {Axis::CHANNELS, "Z"}, {Axis::BATCH, "B"}, in GetAveragePoolingKernelCode()
54 for (auto axis : {Axis::WIDTH, Axis::HEIGHT, Axis::DEPTH, Axis::CHANNELS}) { in GetAveragePoolingKernelCode()
78 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
106 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
133 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
[all …]
Dconcat_xy.cc36 std::map<Axis, std::string> axis_to_selector = { in GetConcatKernelCode()
37 {Axis::WIDTH, "Width"}, {Axis::HEIGHT, "Height"}, in GetConcatKernelCode()
38 {Axis::DEPTH, "Depth"}, {Axis::CHANNELS, "Channels"}, in GetConcatKernelCode()
39 {Axis::BATCH, "Batch"}, in GetConcatKernelCode()
41 std::map<Axis, std::string> axis_to_coord = { in GetConcatKernelCode()
42 {Axis::WIDTH, "X"}, {Axis::HEIGHT, "Y"}, {Axis::DEPTH, "D"}, in GetConcatKernelCode()
43 {Axis::CHANNELS, "S"}, {Axis::BATCH, "B"}, in GetConcatKernelCode()
49 {Axis::WIDTH, Axis::HEIGHT, Axis::DEPTH, Axis::CHANNELS, Axis::BATCH}) { in GetConcatKernelCode()
50 if (op_def.src_tensors[0].HasAxis(axis) && axis != Axis::BATCH) { in GetConcatKernelCode()
72 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetConcatKernelCode()
[all …]
Dreduce.cc48 bool HasAxis(const std::vector<Axis>& axis, Axis a) { in HasAxis()
92 std::map<Axis, int> GetSizesFromShape(const std::set<Axis>& axis, in GetSizesFromShape()
94 std::map<Axis, int> result; in GetSizesFromShape()
101 std::map<Axis, int> GetSizesFromShape(const std::set<Axis>& axis, in GetSizesFromShape()
103 std::map<Axis, int> result; in GetSizesFromShape()
112 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce() argument
115 std::vector<Axis> ordered_axis_to_reduce; in Reduce()
118 {Axis::CHANNELS, Axis::DEPTH, Axis::HEIGHT, Axis::WIDTH, Axis::BATCH}) { in Reduce()
123 if (a == Axis::CHANNELS) { in Reduce()
162 const std::vector<Axis>& axis_to_reduce, in GetReduceKernelCode()
[all …]
Dsplit.cc28 code_ = attr.axis == Axis::CHANNELS ? GetSplitChannelsCode() : GetSplitCode(); in Split()
37 attr_.axis == Axis::WIDTH ? "1" : "args.src_tensor.Width()"; in GetSplitCode()
39 attr_.axis == Axis::HEIGHT ? "1" : "args.src_tensor.Height()"; in GetSplitCode()
41 attr_.axis == Axis::DEPTH ? "1" : "args.src_tensor.Depth()"; in GetSplitCode()
43 attr_.axis == Axis::BATCH ? "1" : "args.src_tensor.Batch()"; in GetSplitCode()
45 attr_.axis == Axis::CHANNELS ? "1" : "args.src_tensor.Slices()"; in GetSplitCode()
51 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitCode()
58 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GetSplitCode()
90 definition_.src_tensors[0].HasAxis(Axis::BATCH) ? ", B" : ""; in GetSplitChannelsCode()
94 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitChannelsCode()
[all …]
Ddepthwise_conv.cc93 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
122 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
141 const std::vector<Axis> axes{Axis::WIDTH, Axis::HEIGHT, Axis::DEPTH}; in GenerateDepthwiseConvolutionCode()
156 const std::vector<Axis> axes{Axis::WIDTH, Axis::HEIGHT, Axis::DEPTH}; in GenerateDepthwiseConvolutionCode()
172 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
175 if (!src_desc.SupportsZeroClamp(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
179 if (op_def.dst_tensors[0].HasAxis(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
182 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
186 if (op_def.dst_tensors[0].HasAxis(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
192 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
[all …]
Dconvolution_transposed.cc200 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
203 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
206 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
220 const std::vector<Axis> axes{Axis::WIDTH, Axis::HEIGHT, Axis::DEPTH}; in GenerateConvolutionTransposedCode()
260 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
283 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
316 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
330 if (!src_def.SupportsZeroClamp(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
333 if (!src_def.CanReadOutOfBorder(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
339 if (block_size.z == 1 && !src_def.SupportsZeroClamp(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
[all …]
Dreduce_test_util.cc32 const std::set<tflite::gpu::Axis> axis{Axis::HEIGHT, Axis::WIDTH}; in MeanHWTest()
59 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceSumChannelsTest()
86 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceProductChannelsTest()
114 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceMaxChannelsTest()
142 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceMinChannelsTest()
Dmax_unpooling.cc49 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
62 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetMaxUnpoolingKernelCode()
73 std::string src_args = op_def.dst_tensors[0].HasAxis(Axis::DEPTH) in GetMaxUnpoolingKernelCode()
77 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
97 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetMaxUnpoolingKernelCode()
104 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
117 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
Dreduce.h33 Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type,
52 const std::vector<Axis>& axis_to_reduce,
58 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape,
62 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce,
/external/apache-xml/src/main/java/org/apache/xpath/axes/
DWalkerFactory.java24 import org.apache.xml.dtm.Axis;
310 return Axis.FOLLOWING; in getAxisFromStep()
312 return Axis.FOLLOWINGSIBLING; in getAxisFromStep()
314 return Axis.PRECEDING; in getAxisFromStep()
316 return Axis.PRECEDINGSIBLING; in getAxisFromStep()
318 return Axis.PARENT; in getAxisFromStep()
320 return Axis.NAMESPACE; in getAxisFromStep()
322 return Axis.ANCESTOR; in getAxisFromStep()
324 return Axis.ANCESTORORSELF; in getAxisFromStep()
326 return Axis.ATTRIBUTE; in getAxisFromStep()
[all …]
DWalkingIteratorSorted.java23 import org.apache.xml.dtm.Axis;
108 boolean isSimpleDownAxis = ((axis == Axis.CHILD) in canBeWalkedInNaturalDocOrderStatic()
109 || (axis == Axis.SELF) in canBeWalkedInNaturalDocOrderStatic()
110 || (axis == Axis.ROOT)); in canBeWalkedInNaturalDocOrderStatic()
120 if(walker.isDocOrdered() && (axis == Axis.DESCENDANT || in canBeWalkedInNaturalDocOrderStatic()
121 axis == Axis.DESCENDANTORSELF || axis == Axis.DESCENDANTSFROMROOT in canBeWalkedInNaturalDocOrderStatic()
122 || axis == Axis.DESCENDANTSORSELFFROMROOT) || (axis == Axis.ATTRIBUTE)) in canBeWalkedInNaturalDocOrderStatic()
DMatchPatternIterator.java23 import org.apache.xml.dtm.Axis;
123 m_superAxis = Axis.ALL; in MatchPatternIterator()
127 m_superAxis = Axis.DESCENDANTSFROMROOT; in MatchPatternIterator()
134 m_superAxis = Axis.ALLFROMNODE; in MatchPatternIterator()
138 m_superAxis = Axis.DESCENDANTORSELF; in MatchPatternIterator()
143 m_superAxis = Axis.ALL; in MatchPatternIterator()
147 System.out.println("axis: "+Axis.getNames(m_superAxis)); in MatchPatternIterator()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConcatenation.h23 template<typename Axis, typename LhsXprType, typename RhsXprType>
24 struct traits<TensorConcatenationOp<Axis, LhsXprType, RhsXprType> >
42 template<typename Axis, typename LhsXprType, typename RhsXprType>
43 struct eval<TensorConcatenationOp<Axis, LhsXprType, RhsXprType>, Eigen::Dense>
45 typedef const TensorConcatenationOp<Axis, LhsXprType, RhsXprType>& type;
48 template<typename Axis, typename LhsXprType, typename RhsXprType>
49 struct nested<TensorConcatenationOp<Axis, LhsXprType, RhsXprType>, 1, typename eval<TensorConcatena…
51 typedef TensorConcatenationOp<Axis, LhsXprType, RhsXprType> type;
57 template<typename Axis, typename LhsXprType, typename RhsXprType>
58 class TensorConcatenationOp : public TensorBase<TensorConcatenationOp<Axis, LhsXprType, RhsXprType>…
[all …]
/external/rust/crates/criterion-plot/src/
Dmap.rs4 use crate::Axis;
13 state: Option<Axis>,
17 type Item = (Axis, &'a T);
19 fn next(&mut self) -> Option<(Axis, &'a T)> { in next() argument
45 pub fn contains_key(&self, key: Axis) -> bool { in contains_key()
49 pub fn get(&self, key: Axis) -> Option<&T> { in get()
53 pub fn get_mut(&mut self, key: Axis) -> Option<&mut T> { in get_mut()
57 pub fn insert(&mut self, key: Axis, value: T) -> Option<T> { in insert() argument
69 state: Some(Axis::BottomX), in iter()
Ddisplay.rs4 use crate::{Axes, Axis, Color, Display, Grid, LineType, PointType, Terminal};
6 impl Display<&'static str> for Axis { implementation
9 Axis::BottomX => "x", in display()
10 Axis::LeftY => "y", in display()
11 Axis::RightY => "y2", in display()
12 Axis::TopX => "x2", in display()
/external/apache-xml/src/main/java/org/apache/xml/dtm/ref/
DDTMDefaultBaseTraversers.java108 m_traversers = new DTMAxisTraverser[Axis.getNamesLength()]; in getAxisTraverser()
121 case Axis.ANCESTOR : in getAxisTraverser()
124 case Axis.ANCESTORORSELF : in getAxisTraverser()
127 case Axis.ATTRIBUTE : in getAxisTraverser()
130 case Axis.CHILD : in getAxisTraverser()
133 case Axis.DESCENDANT : in getAxisTraverser()
136 case Axis.DESCENDANTORSELF : in getAxisTraverser()
139 case Axis.FOLLOWING : in getAxisTraverser()
142 case Axis.FOLLOWINGSIBLING : in getAxisTraverser()
145 case Axis.NAMESPACE : in getAxisTraverser()
[all …]
DDTMDefaultBaseIterators.java120 case Axis.SELF : in getTypedAxisIterator()
123 case Axis.CHILD : in getTypedAxisIterator()
126 case Axis.PARENT : in getTypedAxisIterator()
128 case Axis.ANCESTOR : in getTypedAxisIterator()
130 case Axis.ANCESTORORSELF : in getTypedAxisIterator()
132 case Axis.ATTRIBUTE : in getTypedAxisIterator()
134 case Axis.DESCENDANT : in getTypedAxisIterator()
137 case Axis.DESCENDANTORSELF : in getTypedAxisIterator()
140 case Axis.FOLLOWING : in getTypedAxisIterator()
143 case Axis.PRECEDING : in getTypedAxisIterator()
[all …]
/external/skqp/include/effects/
DSkColorMatrix.h38 enum Axis { enum
43 void setRotate(Axis, SkScalar degrees);
44 void setSinCos(Axis, SkScalar sine, SkScalar cosine);
45 void preRotate(Axis, SkScalar degrees);
46 void postRotate(Axis, SkScalar degrees);
/external/skqp/include/core/
DSkFontArguments.h25 struct Axis { struct
43 SkFontArguments& setAxes(const Axis* axes, int axisCount) { in setAxes()
67 const Axis* getAxes(int* axisCount) const { in getAxes()
69 return reinterpret_cast<const Axis*>(fVariationDesignPosition.coordinates); in getAxes()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc98 if (HasAxis(Axis::WIDTH)) { in GetGPUResources()
106 if (HasAxis(Axis::HEIGHT)) { in GetGPUResources()
109 if (HasAxis(Axis::CHANNELS)) { in GetGPUResources()
113 if (HasAxis(Axis::BATCH)) { in GetGPUResources()
116 if (HasAxis(Axis::DEPTH)) { in GetGPUResources()
186 if (HasAxis(Axis::BATCH)) { in PerformSelector()
276 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in GetLinkingContextFromWriteSelector()
516 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in PerformGetWHOffsetSelector()
754 bool TensorDescriptor::HasAxis(Axis axis) const { in HasAxis()
755 if (axis == Axis::WIDTH || axis == Axis::HEIGHT || axis == Axis::CHANNELS) { in HasAxis()
[all …]

12345678