| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 137 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) { in TrueRank()
151 for (const Shape& shape : parameters) { in MakeProgramShape() local
209 Shape shape = ShapeUtil::MakeShape(element_type, dimensions); in MakeShapeWithSparseLayout() local
217 const Shape& shape) { in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
227 Shape* shape) { in PopulateShape()
241 for (const auto& shape : shapes) { in MakeTupleShape() local
262 /* static */ void ShapeUtil::AppendShapeToTuple(const Shape& shape, in AppendShapeToTuple()
268 /* static */ void ShapeUtil::AppendMajorDimension(int bound, Shape* shape) { in AppendMajorDimension()
275 /* static */ bool ShapeUtil::ElementIsIntegral(const Shape& shape) { in ElementIsIntegral()
279 /* static */ bool ShapeUtil::ElementIsIntegralWithBits(const Shape& shape, in ElementIsIntegralWithBits()
[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()
129 /* static */ int64 IndexUtil::GetDimensionStride(const Shape& shape, in GetDimensionStride()
141 /* static */ bool IndexUtil::IndexInBounds(const Shape& shape, in IndexInBounds()
|
| D | shape_util.h | 180 Shape shape; member
311 static bool IsScalar(const Shape& shape) { in IsScalar()
314 static bool IsEffectiveScalar(const Shape& shape) { in IsEffectiveScalar()
361 static bool IsInitialized(const Shape& shape) { in IsInitialized()
629 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus()
646 const Shape& shape, const IndexIterationSpace& iteration_space, in ForEachIndexWithStatus()
654 static void ForEachIndex(const Shape& shape, absl::Span<const int64> base, in ForEachIndex()
669 static Status ForEachIndexWithStatus(const Shape& shape, in ForEachIndexWithStatus()
679 static void ForEachIndex(const Shape& shape, const FnType& visitor_function) { in ForEachIndex()
692 static void ForEachIndexParallel(const Shape& shape, in ForEachIndexParallel()
|
| D | shape_layout.h | 38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {} in ShapeLayout()
61 const Shape& shape() const { return shape_; } in shape() function
|
| D | layout_util.cc | 115 /* static */ Layout LayoutUtil::GetDefaultLayoutForShape(const Shape& shape) { in GetDefaultLayoutForShape()
142 /* static */ void LayoutUtil::SetToDefaultLayout(Shape* shape) { in SetToDefaultLayout()
160 /* static */ Shape LayoutUtil::GetWithDefaultLayout(const Shape& shape) { in GetWithDefaultLayout()
174 const Shape& shape, bool allow_missing_layouts) { in ValidateLayoutInShape()
206 const Shape& shape) { in ValidateLayoutForShape()
259 /* static */ void LayoutUtil::ClearLayout(Shape* shape) { in ClearLayout()
273 /* static */ bool LayoutUtil::IsDenseArray(const Shape& shape) { in IsDenseArray()
293 /* static */ bool LayoutUtil::IsSparseArray(const Shape& shape) { in IsSparseArray()
306 /* static */ bool LayoutUtil::HasLayout(const Shape& shape) { in HasLayout()
332 const Shape& shape) { in MinorToMajor()
|
| D | layout_util_test.cc | 32 Shape shape = ShapeUtil::MakeShape(element_type, dimensions); in MakeShapeWithLayout() local
40 Shape shape = ShapeUtil::MakeShape(element_type, dimensions); in MakeShapeWithSparseLayout() local
47 Shape shape = in TEST_F() local
263 Shape shape = ShapeUtil::MakeTupleShape( in TEST_F() local
291 Shape shape = ShapeUtil::MakeTupleShape( in TEST_F() local
321 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 3, 4}, {0, 1, 2}); in TEST_F() local
396 Shape shape = ShapeUtil::MakeShapeWithLayout(F32, {2, 3}, {0, 1}); in TEST_F() local
406 Shape shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
423 Shape shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
436 Shape shape = ShapeUtil::MakeTupleShape({}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_v2_op_test.cc | 131 TensorShape shape; in TEST_F() local
147 TensorShape shape; in TEST_F() local
163 TensorShape shape; in TEST_F() local
179 TensorShape shape; in TEST_F() local
195 TensorShape shape; in TEST_F() local
211 TensorShape shape; in TEST_F() local
228 TensorShape shape; in TEST_F() local
244 TensorShape shape; in TEST_F() local
260 TensorShape shape; in TEST_F() local
276 TensorShape shape; in TEST_F() local
[all …]
|
| D | save_op_test.cc | 138 TensorShape shape; in TEST_F() local
157 TensorShape shape; in TEST_F() local
176 TensorShape shape; in TEST_F() local
195 TensorShape shape; in TEST_F() local
214 TensorShape shape; in TEST_F() local
232 TensorShape shape; in TEST_F() local
250 TensorShape shape; in TEST_F() local
268 TensorShape shape; in TEST_F() local
286 TensorShape shape; in TEST_F() local
304 TensorShape shape; in TEST_F() local
[all …]
|
| /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/lite/experimental/swift/Tests/ |
| D | TensorTests.swift | 25 let shape = TensorShape(Constant.dimensions) in testTensor_Init() variable
45 let shape = TensorShape(Constant.dimensions) in testTensorShape_InitWithArray() variable
51 let shape = TensorShape(2, 2, 3) in testTensorShape_InitWithElements() variable
57 let shape: TensorShape = [2, 2, 3] in testTensorShape_InitWithArrayLiteral() variable
|
| /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/xla/service/ |
| D | hlo_instructions.cc | 74 HloOpcode opcode, const Shape& shape, HloInstruction* operand, in HloBatchNormInstruction()
106 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormTrainingInstruction()
115 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
124 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormInferenceInstruction()
136 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
145 const Shape& shape, HloInstruction* operand, HloInstruction* scale, in HloBatchNormGradInstruction()
157 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
165 HloFftInstruction::HloFftInstruction(const Shape& shape, in HloFftInstruction()
198 const Shape& shape, absl::Span<HloInstruction* const> new_operands, in CloneWithNewOperandsImpl()
205 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()
|
| D | reduce_precision_insertion_test.cc | 49 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
76 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
103 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
135 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
163 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
200 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
225 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
266 Shape shape = ShapeUtil::MakeShape(S32, {4}); in TEST_F() local
292 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
318 Shape shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
[all …]
|
| D | hlo_instruction.cc | 133 Shape shape(proto.shape()); in CreateFromProto() local
638 int64 parameter_number, const Shape& shape, const string& name) { in CreateParameter()
654 const Shape& shape, int64 iota_dimension) { in CreateIota()
659 HloInstruction::CreateGetTupleElement(const Shape& shape, in CreateGetTupleElement()
666 const Shape& shape, RandomDistribution distribution, in CreateRng()
672 const Shape& shape, HloOpcode opcode, in CreateNary()
686 const Shape& shape, HloOpcode opcode, HloInstruction* operand) { in CreateUnary()
721 const Shape& shape, HloOpcode opcode, HloInstruction* lhs, in CreateBinary()
751 const Shape& shape, HloOpcode opcode, HloInstruction* lhs, in CreateTernary()
768 const Shape& shape, HloOpcode opcode, in CreateVariadic()
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | datasets_test.py | 163 def fn(dataset, tensor, shape): argument
171 def fn(dataset, tensor, shape): argument
179 def fn(dataset, tensor, shape): argument
187 def fn(dataset, tensor, shape): argument
195 def fn(dataset, tensor, shape): argument
203 def fn(dataset, tensor, shape): argument
211 def fn(dataset, tensor, shape): argument
219 def fn(dataset, tensor, shape): argument
227 def fn(dataset, tensor, shape): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | random_ops.py | 38 def _ShapeTensor(shape): argument
49 def random_normal(shape, argument
88 def parameterized_truncated_normal(shape, argument
144 def truncated_normal(shape, argument
190 def random_uniform(shape, argument
401 def random_gamma(shape, argument
485 def random_poisson(lam, shape, dtype=dtypes.float32, seed=None, name=None): argument
524 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
96 def __call__(self, shape, dtype=dtypes.float32): argument
105 def __call__(self, shape, dtype=dtypes.float32): argument
197 def __call__(self, shape, dtype=None): argument
238 def __call__(self, shape, dtype=dtypes.float32): argument
283 def __call__(self, shape, dtype=dtypes.float32): argument
330 def __call__(self, shape, dtype=dtypes.float32): argument
401 def __call__(self, shape, dtype=dtypes.float32): argument
475 def __call__(self, shape, dtype=dtypes.float32): argument
527 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/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/training/ |
| D | warm_starting_util_test.py | 59 shape=None, argument
540 partitioner = lambda shape, dtype: [1] * len(shape) argument
569 partitioner = lambda shape, dtype: [1] * len(shape) argument
603 partitioner = lambda shape, dtype: [1] * len(shape) argument
639 partitioner = lambda shape, dtype: [1] * len(shape) argument
684 partitioner = lambda shape, dtype: [1] * len(shape) argument
727 partitioner = lambda shape, dtype: [1] * len(shape) argument
798 partitioner = lambda shape, dtype: [1] * len(shape) argument
870 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
996 def _partitioner(shape, dtype): # pylint:disable=unused-argument argument
[all …]
|
| /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/java/src/main/java/org/tensorflow/ |
| D | Shape.java | 55 long[] shape = new long[otherDimensionSizes.length + 1]; in make() local
108 Shape(long[] shape) { in Shape()
118 private long[] shape; field in Shape
|
| /external/u-boot/drivers/demo/ |
| D | demo-shape.c | 32 static const struct shape { in shape_hello() struct
33 int start; in shape_hello()
34 int end; in shape_hello()
35 int dstart; in shape_hello()
36 int dend; in shape_hello()
42 struct shape shape; in shape_hello() local
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | utils.py | 68 def check_default_value(shape, default_value, dtype, key): argument
127 def _create_tuple(shape, value): argument
140 def _is_shape_and_default_value_compatible(default_value, shape): argument
|