| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ElementwiseUnaryFixture.h | 127 TensorType compute_target(const TensorShape &shape, DataType data_type, bool in_place) in compute_target()
179 SimpleTensor<T> compute_reference(const TensorShape &shape, DataType data_type) in compute_reference()
201 void setup(const TensorShape &shape, DataType data_type) in setup()
212 void setup(const TensorShape &shape, DataType data_type) in setup()
223 void setup(const TensorShape &shape, DataType data_type) in setup()
234 void setup(const TensorShape &shape, DataType data_type) in setup()
245 void setup(const TensorShape &shape, DataType data_type, bool in_place) in setup()
256 void setup(const TensorShape &shape, DataType data_type) in setup()
267 void setup(const TensorShape &shape, DataType data_type) in setup()
278 void setup(const TensorShape &shape, DataType data_type) in setup()
[all …]
|
| D | ScaleFixture.h | 42 …void setup(TensorShape shape, DataType data_type, QuantizationInfo quantization_info, DataLayout d… in setup()
81 void generate_scale(const TensorShape &shape) in generate_scale()
133 TensorType compute_target(TensorShape shape, DataLayout data_layout) in compute_target()
183 SimpleTensor<T> compute_reference(const TensorShape &shape) in compute_reference()
215 …void setup(TensorShape shape, DataType data_type, QuantizationInfo quantization_info, DataLayout d… in setup()
235 …void setup(TensorShape shape, DataType data_type, QuantizationInfo input_quantization_info, Quanti… in setup()
256 …void setup(TensorShape shape, DataType data_type, DataLayout data_layout, InterpolationPolicy poli… in setup()
|
| D | SoftmaxLayerFixture.h | 50 …void setup(TensorShape shape, DataType data_type, QuantizationInfo quantization_info, float beta, … in setup()
83 TensorType compute_target(const TensorShape &shape, DataType data_type, in compute_target()
113 SimpleTensor<T> compute_reference(const TensorShape &shape, DataType data_type, in compute_reference()
135 void setup(TensorShape shape, DataType data_type, float beta, size_t axis) in setup()
150 …void setup(TensorShape shape, DataType data_type, QuantizationInfo quantization_info, float beta, … in setup()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 144 for (const auto& shape : shapes) { in MakeTupleShapeImpl() local
199 /* static */ int64_t ShapeUtil::TrueRank(const Shape& shape) { in TrueRank()
212 Shape* shape) { in FillNewShape()
248 for (const Shape& shape : parameters) { in MakeProgramShape() local
257 Shape shape; in MakeShape() local
274 const Shape& shape) { in MakeShapeWithStaticDimensions()
282 Shape shape; in MakeValidatedShape() local
300 Shape shape; in MakeValidatedShape() local
325 /* static */ Shape ShapeUtil::MoveDimToMajor(const Shape& shape, int64_t dim) { in MoveDimToMajor()
360 const Shape& shape) { in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
[all …]
|
| D | shape_util.h | 102 Shape shape; member
239 static bool IsScalar(const Shape& shape) { in IsScalar()
242 static bool IsEffectiveScalar(const Shape& shape) { in IsEffectiveScalar()
312 static bool IsInitialized(const Shape& shape) { in IsInitialized()
626 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus()
648 const Shape& shape, const IndexIterationSpace& iteration_space, in ForEachIndexWithStatus()
656 static void ForEachIndex(const Shape& shape, absl::Span<const int64_t> base, in ForEachIndex()
671 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus()
681 static void ForEachIndex(const Shape& shape, const FnType& visitor_function) { in ForEachIndex()
694 static void ForEachIndexParallel(const Shape& shape, in ForEachIndexParallel()
[all …]
|
| D | layout_util.cc | 118 /* static */ Layout LayoutUtil::GetDefaultLayoutForShape(const Shape& shape) { in GetDefaultLayoutForShape()
145 /* static */ void LayoutUtil::SetToDefaultLayout(Shape* shape) { in SetToDefaultLayout()
162 /* static */ Shape LayoutUtil::GetWithDefaultLayout(const Shape& shape) { in GetWithDefaultLayout()
176 const Shape& shape, bool allow_missing_layouts) { in ValidateLayoutInShape()
208 const Shape& shape) { in ValidateLayoutForShape()
271 /* static */ void LayoutUtil::ClearLayout(Shape* shape) { in ClearLayout()
285 /* static */ void LayoutUtil::ClearTiles(Shape* shape) { in ClearTiles()
296 /* static */ bool LayoutUtil::IsDenseArray(const Shape& shape) { in IsDenseArray()
300 /* static */ bool LayoutUtil::IsSparseArray(const Shape& shape) { in IsSparseArray()
304 /* static */ bool LayoutUtil::IsCOOArray(const Shape& shape) { in IsCOOArray()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | signed_quantized_split_test.cc | 38 const std::vector<int32_t> shape({shape_rng() * 2}); variable
64 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
90 std::vector<int32_t> shape({shape_rng(), shape_rng(), shape_rng()}); variable
116 std::vector<int32_t> shape( variable
139 const std::vector<int32_t> shape({shape_rng() * 3}); variable
165 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
191 std::vector<int32_t> shape({shape_rng(), shape_rng(), shape_rng()}); variable
217 std::vector<int32_t> shape( variable
240 const std::vector<int32_t> shape({shape_rng() * 4}); variable
266 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
[all …]
|
| D | split_test.cc | 38 const std::vector<int32_t> shape({shape_rng() * 2}); variable
64 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
90 std::vector<int32_t> shape({shape_rng(), shape_rng(), shape_rng()}); variable
116 std::vector<int32_t> shape( variable
139 const std::vector<int32_t> shape({shape_rng() * 3}); variable
165 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
191 std::vector<int32_t> shape({shape_rng(), shape_rng(), shape_rng()}); variable
217 std::vector<int32_t> shape( variable
240 const std::vector<int32_t> shape({shape_rng() * 4}); variable
266 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
[all …]
|
| D | unsigned_quantized_split_test.cc | 38 const std::vector<int32_t> shape({shape_rng() * 2}); variable
64 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
90 std::vector<int32_t> shape({shape_rng(), shape_rng(), shape_rng()}); variable
116 std::vector<int32_t> shape( variable
139 const std::vector<int32_t> shape({shape_rng() * 3}); variable
165 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
191 std::vector<int32_t> shape({shape_rng(), shape_rng(), shape_rng()}); variable
217 std::vector<int32_t> shape( variable
240 const std::vector<int32_t> shape({shape_rng() * 4}); variable
266 std::vector<int32_t> shape({shape_rng(), shape_rng()}); variable
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_v2_op_test.cc | 133 TensorShape shape; in TEST_F() local
149 TensorShape shape; in TEST_F() local
165 TensorShape shape; in TEST_F() local
181 TensorShape shape; in TEST_F() local
197 TensorShape shape; in TEST_F() local
213 TensorShape shape; in TEST_F() local
230 TensorShape shape; in TEST_F() local
246 TensorShape shape; in TEST_F() local
262 TensorShape shape; in TEST_F() local
278 TensorShape shape; in TEST_F() local
[all …]
|
| D | save_op_test.cc | 141 TensorShape shape; in TEST_F() local
160 TensorShape shape; in TEST_F() local
179 TensorShape shape; in TEST_F() local
198 TensorShape shape; in TEST_F() local
217 TensorShape shape; in TEST_F() local
235 TensorShape shape; in TEST_F() local
253 TensorShape shape; in TEST_F() local
271 TensorShape shape; in TEST_F() local
289 TensorShape shape; in TEST_F() local
307 TensorShape shape; in TEST_F() local
[all …]
|
| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/tensorbuffer/ |
| D | TensorBuffer.java | 34 protected int[] shape; field in TensorBuffer
75 public static TensorBuffer createFixedSize(@NonNull int[] shape, DataType dataType) { in createFixedSize()
245 public abstract void loadArray(@NonNull int[] src, @NonNull int[] shape); in loadArray()
277 public abstract void loadArray(@NonNull float[] src, @NonNull int[] shape); in loadArray()
306 public void loadBuffer(@NonNull ByteBuffer buffer, @NonNull int[] shape) { in loadBuffer()
337 protected TensorBuffer(@NonNull int[] shape) { in TensorBuffer()
350 protected static int computeFlatSize(@NonNull int[] shape) { in computeFlatSize()
363 protected void resize(@NonNull int[] shape) { in resize()
380 private void allocateMemory(@NonNull int[] shape) { in allocateMemory()
415 private static boolean isShapeValid(@NonNull int[] shape) { in isShapeValid()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 112 HloOpcode opcode, const Shape& shape, HloInstruction* operand, in HloBatchNormInstruction()
144 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormTrainingInstruction()
153 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
162 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormInferenceInstruction()
174 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
183 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormGradInstruction()
195 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
203 HloFftInstruction::HloFftInstruction(const Shape& shape, in HloFftInstruction()
236 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
244 HloOpcode opcode, const Shape& shape, in HloAsyncInstruction()
[all …]
|
| D | indexed_array_analysis.h | 78 const Shape& shape() const override { return instruction().shape(); } in shape() function
95 const Shape& shape() const override { return literal()->shape(); } in shape() function
114 const Shape& shape() const override { return shape_; } in shape() function
117 explicit ReshapedArray(Array* operand, Shape shape) in ReshapedArray()
178 const Shape& shape() const override { return shape_; } in shape() function
196 std::vector<int64_t> output_dims, Shape shape) in ScalarIndexedArray()
228 Shape shape) in ScalarIndexedConstantArray()
339 std::vector<int64_t> output_dims, Shape shape) { in ConstructScalarIndexedArray()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 81 auto shape = std::make_unique<Shape>(TypeToShape(ty)); in MakeXlaOp() local
101 const Shape& shape, XlaOp lhs, XlaOp rhs, const Window& window, in ConvGeneralDilatedInternal()
128 const Shape& shape, XlaOp operand, FftType fft_type, in FftInternal()
144 const XlaComputation* computation, const Shape& shape, in CustomCallInternal()
225 const Shape& shape, absl::Span<const XlaOp> all_operands, in ReduceInternal()
243 const Shape& shape, XlaOp operand, XlaOp init_value, in ReduceWindowInternal()
272 XlaOp MlirHloBuilder::Iota(const Shape& shape, int64_t iota_dimension) { in Iota()
284 StatusOr<XlaOp> MlirHloBuilder::BitcastConvertTypeInternal(const Shape& shape, in BitcastConvertTypeInternal()
294 const Shape& shape, XlaOp operand, absl::Span<const int64_t> permutation) { in TransposeInternal()
303 const Shape& shape, XlaOp operand, absl::Span<const int64_t> dimensions) { in RevInternal()
[all …]
|
| /external/ComputeLibrary/tests/datasets/ |
| D | ScaleValidationDataset.h | 170 #define ASSEMBLE_DATASET(shape, samping_policy_set) \ argument
176 #define ASSEMBLE_DATASET_DYNAMIC_FUSION(shape, samping_policy_set) … argument
181 #define ASSEMBLE_S8_DATASET(shape, samping_policy_set) … argument
187 #define ASSEMBLE_NHWC_DATASET(shape, samping_policy_set) … argument
194 #define ASSEMBLE_QUANTIZED_DATASET(shape, sampling_policy_set, quantization_info_set) \ argument
202 #define ASSEMBLE_QUANTIZED_DATASET_DYNAMIC_FUSION(shape, sampling_policy_set, quantization_info_set… argument
210 #define ASSEMBLE_DIFFERENTLY_QUANTIZED_DATASET(shape, sampling_policy_set, input_quant_info_set, ou… argument
|
| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/image/ |
| D | ColorSpaceType.java | 44 int[] getNormalizedShape(int[] shape) { in getNormalizedShape()
86 int[] getNormalizedShape(int[] shape) { in getNormalizedShape()
143 void assertShape(int[] shape) { in assertShape()
162 int getWidth(int[] shape) { in getWidth()
172 int getHeight(int[] shape) { in getHeight()
183 abstract int[] getNormalizedShape(int[] shape); in getNormalizedShape()
203 protected boolean isValidNormalizedShape(int[] shape) { in isValidNormalizedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | matmul_utils_test.cc | 33 Shape shape = ParseShape("f32[1,2,3,4,5,6]").ValueOrDie(); in TEST() local
40 absl::string_view shape; member
53 Shape shape = ParseShape(params.shape).ValueOrDie(); in TEST_P() local
76 Shape shape = ParseShape("f32[3,4,5,6,7,8]{5,4,3,2,1,0}").ValueOrDie(); in TEST() local
84 Shape shape = ParseShape("f32[3,4,5,6]{3,2,0,1}").ValueOrDie(); in TEST() local
93 absl::string_view shape; member
107 Shape shape = ParseShape(params.shape).ValueOrDie(); in TEST_P() local
129 Shape shape = ParseShape("f32[3,4,5]{0,2,1}").ValueOrDie(); in TEST() local
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/core/ |
| D | ZerosTest.java | 42 long[] shape = {2, 2}; in createIntZeros() local
60 long[] shape = {2, 2}; in createFloatZeros() local
78 long[] shape = {2, 2}; in createDoubleZeros() local
96 long[] shape = {2, 2}; in createLongZeros() local
114 long[] shape = {2, 2}; in createBooleanZeros() local
132 long[] shape = {2, 2}; in createUInt8Zeros() local
151 long[] shape = {2, 2}; in cannotCreateStringZeros() local
162 long[] shape = {2, 2}; in operationsComposingZerosAreCorrectlyNamed() local
|
| /external/tensorflow/tensorflow/lite/swift/Tests/ |
| D | TensorTests.swift | 24 let shape = Tensor.Shape(Constant.dimensions) in testInit() variable
44 let shape = Tensor.Shape(Constant.dimensions) in testEquatable() variable
77 let shape = Tensor.Shape(Constant.dimensions) in testInitWithArray() variable
83 let shape = Tensor.Shape(2, 2, 3) in testInitWithElements() variable
89 let shape: Tensor.Shape = [2, 2, 3] in testInitWithArrayLiteral() variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | elementwise_test.cc | 34 TensorRef<BHWC> GetTensorRef(int ref, const BHWC& shape) { in GetTensorRef()
44 const BHWC shape(1, 2, 2, 1); in TEST() local
56 const BHWC shape(1, 2, 2, 1); in TEST() local
68 const BHWC shape(1, 2, 2, 1); in TEST() local
79 const BHWC shape(1, 1, 1, 7); in TEST() local
94 const BHWC shape(1, 1, 1, 7); in TEST() local
110 const BHWC shape(1, 1, 1, 7); in TEST() local
124 const BHWC shape(1, 1, 1, 7); in TEST() local
138 const BHWC shape(1, 2, 2, 1); in TEST() local
150 const BHWC shape(1, 2, 2, 1); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | testing.cc | 35 int64_t DataSizeOfShape(const Shape& shape) { in DataSizeOfShape()
48 XlaOp BuildFakeDataOpOnDevice(const Shape& shape, XlaBuilder* builder) { in BuildFakeDataOpOnDevice()
64 const Shape& shape, Client* client, DebugOptions* debug_opts) { in MakeFakeDataViaDeviceOrDie()
81 const Shape& shape, Client* client, DebugOptions* debug_opts /*=nullptr*/) { in MakeFakeDataOrDie()
106 for (const ShapeProto& shape : program_shape.parameters()) { in MakeFakeArgumentsOrDie() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/attributes/ |
| D | ReshapeAttributes.cpp | 33 ReshapeAttributes &ReshapeAttributes::shape(const TensorShape &shape) in shape() function in arm_compute::experimental::dynamic_fusion::ReshapeAttributes
38 TensorShape ReshapeAttributes::shape() const in shape() function in arm_compute::experimental::dynamic_fusion::ReshapeAttributes
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 56 shape=None, argument
564 partitioner = lambda shape, dtype: [1] * len(shape) argument
593 partitioner = lambda shape, dtype: [1] * len(shape) argument
627 partitioner = lambda shape, dtype: [1] * len(shape) argument
663 partitioner = lambda shape, dtype: [1] * len(shape) argument
708 partitioner = lambda shape, dtype: [1] * len(shape) argument
751 partitioner = lambda shape, dtype: [1] * len(shape) argument
822 partitioner = lambda shape, dtype: [1] * len(shape) argument
894 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
1020 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | datasets_test.py | 160 def fn(dataset, tensor, shape): argument
168 def fn(dataset, tensor, shape): argument
176 def fn(dataset, tensor, shape): argument
184 def fn(dataset, tensor, shape): argument
192 def fn(dataset, tensor, shape): argument
200 def fn(dataset, tensor, shape): argument
208 def fn(dataset, tensor, shape): argument
216 def fn(dataset, tensor, shape): argument
224 def fn(dataset, tensor, shape): argument
|