/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | shape.h | 32 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 …]
|
D | shape_test.cc | 32 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 …]
|
D | shape.cc | 31 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 …]
|
D | operations.cc | 276 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 …]
|
D | model_builder_helper.cc | 123 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/ |
D | pooling.cc | 42 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 …]
|
D | concat_xy.cc | 36 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 …]
|
D | reduce.cc | 48 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 …]
|
D | split.cc | 28 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 …]
|
D | depthwise_conv.cc | 93 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 …]
|
D | convolution_transposed.cc | 200 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 …]
|
D | reduce_test_util.cc | 32 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()
|
D | max_unpooling.cc | 49 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()
|
D | reduce.h | 33 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/ |
D | WalkerFactory.java | 24 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 …]
|
D | WalkingIteratorSorted.java | 23 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()
|
D | MatchPatternIterator.java | 23 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/ |
D | TensorConcatenation.h | 23 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/ |
D | map.rs | 4 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()
|
D | display.rs | 4 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/ |
D | DTMDefaultBaseTraversers.java | 108 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 …]
|
D | DTMDefaultBaseIterators.java | 120 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/ |
D | SkColorMatrix.h | 38 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/ |
D | SkFontArguments.h | 25 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/ |
D | tensor_desc.cc | 98 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 …]
|