| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 153 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) { in TrueRank()
167 for (const Shape& shape : parameters) { in MakeProgramShape() local
191 const Shape& shape) { in MakeShapeWithStaticDimensions()
234 const Shape& shape) { in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
254 Shape* shape) { in PopulateShape()
268 for (const auto& shape : shapes) { in MakeTupleShape() local
289 /* static */ void ShapeUtil::AppendShapeToTuple(const Shape& shape, in AppendShapeToTuple()
295 /* static */ void ShapeUtil::UpdateTupleShape(const Shape& shape, int64 index, in UpdateTupleShape()
301 /* static */ void ShapeUtil::UpdateDynamicDimension(Shape* shape, in UpdateDynamicDimension()
315 /* static */ void ShapeUtil::AppendMajorDimension(int bound, Shape* shape) { in AppendMajorDimension()
[all …]
|
| D | index_util.cc | 29 const Shape& shape, absl::Span<const int64> multi_index) { in MultidimensionalIndexToLinearIndex()
93 const Shape& shape, int64 linear_index) { in LinearIndexToMultidimensionalIndex()
116 /* static */ bool IndexUtil::BumpIndices(const Shape& shape, in BumpIndices()
131 /* static */ int64 IndexUtil::GetDimensionStride(const Shape& shape, in GetDimensionStride()
143 /* static */ bool IndexUtil::IndexInBounds(const Shape& shape, in IndexInBounds()
|
| D | shape_util.h | 191 Shape shape; member
313 static bool IsScalar(const Shape& shape) { in IsScalar()
316 static bool IsEffectiveScalar(const Shape& shape) { in IsEffectiveScalar()
371 static bool IsInitialized(const Shape& shape) { in IsInitialized()
656 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus()
673 const Shape& shape, const IndexIterationSpace& iteration_space, in ForEachIndexWithStatus()
681 static void ForEachIndex(const Shape& shape, absl::Span<const int64> base, in ForEachIndex()
696 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus()
706 static void ForEachIndex(const Shape& shape, const FnType& visitor_function) { in ForEachIndex()
719 static void ForEachIndexParallel(const Shape& shape, in ForEachIndexParallel()
|
| D | layout_util_test.cc | 32 Shape shape = ShapeUtil::MakeShape(element_type, dimensions); in MakeShapeWithLayout() local
39 Shape shape = in TEST_F() local
197 Shape shape = ShapeUtil::MakeTupleShape( in TEST_F() local
225 Shape shape = ShapeUtil::MakeTupleShape( in TEST_F() local
255 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 3, 4}, {0, 1, 2}); in TEST_F() local
330 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 3}, {0, 1}); in TEST_F() local
340 Shape shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
357 Shape shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
370 Shape shape = ShapeUtil::MakeTupleShape({}); in TEST_F() local
393 Shape shape = ShapeUtil::MakeTupleShape({sub_1, sub_2}); in TEST_F() local
|
| D | shape_layout.h | 38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {} in ShapeLayout()
62 const Shape& shape() const { return shape_; } in shape() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | elementwise_test.cc | 31 TensorRef<BHWC> GetTensorRef(int ref, const BHWC& shape) { in GetTensorRef()
41 const BHWC shape(1, 2, 2, 1); in TEST() local
53 const BHWC shape(1, 2, 2, 1); in TEST() local
65 const BHWC shape(1, 2, 2, 1); in TEST() local
79 const BHWC shape(1, 1, 1, 7); in TEST() local
93 const BHWC shape(1, 2, 2, 1); in TEST() local
105 const BHWC shape(1, 2, 2, 1); in TEST() local
119 const BHWC shape(1, 2, 2, 1); in TEST() local
131 const BHWC shape(1, 2, 2, 1); in TEST() local
143 const BHWC shape(1, 2, 2, 1); in TEST() local
[all …]
|
| D | converter_test.cc | 44 Dimensions ToDimensions(const BHWC& shape) { in ToDimensions()
48 Status RunFromTensorTest(const BHWC& shape) { in RunFromTensorTest()
97 BHWC shape(1, h, w, c); in TEST() local
106 Status RunToTensorTest(const BHWC& shape) { in RunToTensorTest()
154 BHWC shape(1, h, w, c); in TEST() local
|
| /external/tensorflow/tensorflow/lite/experimental/support/java/src/java/org/tensorflow/lite/support/tensorbuffer/ |
| D | TensorBuffer.java | 31 protected int[] shape; field in TensorBuffer
72 public static TensorBuffer createFixedSize(@NonNull int[] shape, DataType dataType) { in createFixedSize()
195 public abstract void loadArray(@NonNull int[] src, @NonNull int[] shape); in loadArray()
227 public abstract void loadArray(@NonNull float[] src, @NonNull int[] shape); in loadArray()
256 public void loadBuffer(@NonNull ByteBuffer buffer, @NonNull int[] shape) { in loadBuffer()
295 protected TensorBuffer(@NonNull int[] shape) { in TensorBuffer()
308 protected static int computeFlatSize(@NonNull int[] shape) { in computeFlatSize()
321 protected void resize(@NonNull int[] shape) { in resize()
338 private void allocateMemory(@NonNull int[] shape) { in allocateMemory()
360 private static boolean isShapeValid(@NonNull int[] shape) { in isShapeValid()
|
| D | TensorBufferUint8.java | 32 TensorBufferUint8(@NonNull int[] shape) { in TensorBufferUint8()
75 public void loadArray(@NonNull float[] src, @NonNull int[] shape) { in loadArray()
89 public void loadArray(@NonNull int[] src, @NonNull int[] shape) { in loadArray()
|
| /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/tensorflow/tensorflow/lite/experimental/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/compiler/xla/client/lib/ |
| D | testing.cc | 36 int64 DataSizeOfShape(const Shape& shape) { in DataSizeOfShape()
49 XlaOp BuildFakeDataOpOnDevice(const Shape& shape, XlaBuilder* builder) { in BuildFakeDataOpOnDevice()
63 const Shape& shape, Client* client, DebugOptions* debug_opts) { in MakeFakeDataViaDeviceOrDie()
80 const Shape& shape, Client* client, DebugOptions* debug_opts /*=nullptr*/) { in MakeFakeDataOrDie()
105 for (const ShapeProto& shape : program_shape.parameters()) { in MakeFakeArgumentsOrDie() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 75 HloOpcode opcode, const Shape& shape, HloInstruction* operand, in HloBatchNormInstruction()
107 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormTrainingInstruction()
116 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
125 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormInferenceInstruction()
137 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
146 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormGradInstruction()
158 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
166 HloFftInstruction::HloFftInstruction(const Shape& shape, in HloFftInstruction()
199 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
206 HloCompareInstruction::HloCompareInstruction(const Shape& shape, in HloCompareInstruction()
[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
195 std::vector<int64> output_dims, Shape shape) in ScalarIndexedArray()
227 Shape shape) in ScalarIndexedConstantArray()
338 std::vector<int64> output_dims, Shape shape) { in ConstructScalarIndexedArray()
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/core/ |
| D | ZerosTest.java | 41 long[] shape = {2, 2}; in createIntZeros() local
59 long[] shape = {2, 2}; in createFloatZeros() local
77 long[] shape = {2, 2}; in createDoubleZeros() local
95 long[] shape = {2, 2}; in createLongZeros() local
113 long[] shape = {2, 2}; in createBooleanZeros() local
131 long[] shape = {2, 2}; in createUInt8Zeros() local
150 long[] shape = {2, 2}; in cannotCreateStringZeros() local
160 long[] shape = {2, 2}; in operationsComposingZerosAreCorrectlyNamed() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.h | 36 static StatusOr<TypeT> ConvertTensorShapeToType(const Shape& shape, in ConvertTensorShapeToType()
69 const Shape& shape, mlir::Builder builder) { in ConvertTensorShapeToType()
74 static StatusOr<mlir::Type> ConvertShapeToType(const Shape& shape, in ConvertShapeToType()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | datasets_test.py | 164 def fn(dataset, tensor, shape): argument
172 def fn(dataset, tensor, shape): argument
180 def fn(dataset, tensor, shape): argument
188 def fn(dataset, tensor, shape): argument
196 def fn(dataset, tensor, shape): argument
204 def fn(dataset, tensor, shape): argument
212 def fn(dataset, tensor, shape): argument
220 def fn(dataset, tensor, shape): argument
228 def fn(dataset, tensor, shape): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 61 shape=None, argument
542 partitioner = lambda shape, dtype: [1] * len(shape) argument
571 partitioner = lambda shape, dtype: [1] * len(shape) argument
605 partitioner = lambda shape, dtype: [1] * len(shape) argument
641 partitioner = lambda shape, dtype: [1] * len(shape) argument
686 partitioner = lambda shape, dtype: [1] * len(shape) argument
729 partitioner = lambda shape, dtype: [1] * len(shape) argument
800 partitioner = lambda shape, dtype: [1] * len(shape) argument
872 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
998 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 28 Status PopulateInfeedLayoutVector(const xla::Shape& shape, in PopulateInfeedLayoutVector()
74 xla::Shape* shape) { in AssignLayout()
87 Status XLAShapeToTensorShape(const xla::Shape& shape, in XLAShapeToTensorShape()
103 xla::Shape* shape) { in TensorShapeToXLAShape()
124 xla::StatusOr<std::vector<int>> GetShapeLayoutVector(const xla::Shape& shape) { in GetShapeLayoutVector()
140 const xla::Shape& shape = in GetShapeWithLayout() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | random_ops.py | 44 def random_normal(shape, argument
102 def parameterized_truncated_normal(shape, argument
159 def truncated_normal(shape, argument
206 def random_uniform(shape, argument
458 def _maybe_set_static_shape_helper(tensor, shape, postfix_tensor): argument
470 def random_gamma(shape, argument
563 def random_poisson(lam, shape, dtype=dtypes.float32, seed=None, name=None): argument
602 def random_poisson_v2(shape, lam, dtype=dtypes.float32, seed=None, name=None): argument
|
| D | init_ops_v2.py | 51 def __call__(self, shape, dtype=None): argument
117 def __call__(self, shape, dtype=dtypes.float32): argument
157 def __call__(self, shape, dtype=dtypes.float32): argument
245 def __call__(self, shape, dtype=None): argument
305 def __call__(self, shape, dtype=dtypes.float32): argument
369 def __call__(self, shape, dtype=dtypes.float32): argument
434 def __call__(self, shape, dtype=dtypes.float32): argument
525 def __call__(self, shape, dtype=dtypes.float32): argument
616 def __call__(self, shape, dtype=dtypes.float32): argument
686 def __call__(self, shape, dtype=dtypes.float32): argument
[all …]
|
| D | partitioned_variables.py | 111 def _partitioner(shape, dtype): argument
181 def _partitioner(shape, dtype): argument
233 def _partitioner(shape, **unused_args): argument
245 shape, slicing, initializer, dtype=dtypes.float32, argument
|
| /external/gemmlowp/test/ |
| D | benchmark_meta_gemm.cc | 179 void time_one(Shape* shape, double max_time) { in time_one()
270 for (auto& shape : googlenet_gemms) { in main() local
279 for (auto& shape : small_gemms) { in main() local
288 for (auto& shape : others) { in main() local
299 for (auto& shape : lstm) { in main() local
315 for (auto& shape : googlenet_gemms) { in main() local
319 for (auto& shape : small_gemms) { in main() local
323 for (auto& shape : others) { in main() local
327 for (auto& shape : lstm) { in main() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.cc | 33 Status ConvertToPHWO4I4(absl::Span<const float> in, const OHWI& shape, in ConvertToPHWO4I4()
77 uint32_t GetElementsSizeForPHWO4I4(const OHWI& shape) { in GetElementsSizeForPHWO4I4()
82 uint32_t GetElementsSizeForPHWO4I4(const IHWO& shape) { in GetElementsSizeForPHWO4I4()
107 uint3 Get3DSizeForPHWO4I4(const OHWI& shape) { in Get3DSizeForPHWO4I4()
113 Status ConvertToPHWO4I4(absl::Span<const float> in, const IHWO& shape, in ConvertToPHWO4I4()
162 uint32_t GetElementsSizeForPIOHW4(const OHWI& shape) { in GetElementsSizeForPIOHW4()
167 Status ConvertToPIOHW4(absl::Span<const float> in, const OHWI& shape, in ConvertToPIOHW4()
210 Status ValidateConvertToPHWC4(absl::Span<const float> in, const BHWC& shape, in ValidateConvertToPHWC4()
226 Status ConvertToPHWC4(absl::Span<const float> in, const BHWC& shape, in ConvertToPHWC4()
279 Status ConvertToPHWC4Half(absl::Span<const float> in, const BHWC& shape, in ConvertToPHWC4Half()
[all …]
|