| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 59 def _as_shape_list(shapes, argument
66 if (not isinstance(shapes, collections_abc.Sequence) or not shapes or
67 any(shape is None or isinstance(shape, int) for shape in shapes)):
70 if shapes is None:
72 if isinstance(shapes, tensor_shape.TensorShape):
73 shapes = [shapes]
74 if not isinstance(shapes, (tuple, list)):
77 if all(shape is None or isinstance(shape, int) for shape in shapes):
79 shapes = [shapes]
80 shapes = [tensor_shape.as_shape(shape) for shape in shapes]
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_cross_op_test.py | 516 inds, vals, shapes = gen_sparse_ops.sparse_cross_v2(
519 shapes=[sp_inp_1.dense_shape, sp_inp_2.dense_shape],
522 out = sparse_tensor.SparseTensor(inds, vals, shapes)
542 inds, vals, shapes = gen_sparse_ops.sparse_cross_v2(
545 shapes=[sp_inp_1.dense_shape, sp_inp_2.dense_shape],
548 out = sparse_tensor.SparseTensor(inds, vals, shapes)
570 inds, vals, shapes = gen_sparse_ops.sparse_cross_v2(
573 shapes=[],
576 out = sparse_tensor.SparseTensor(inds, vals, shapes)
598 inds, vals, shapes = gen_sparse_ops.sparse_cross_v2(
[all …]
|
| D | cholesky_op_test.py | 206 shapes = self.getShapes([1, 2, 10])
208 shapes, dtypes=(dtypes_lib.float32, dtypes_lib.float64))
213 shapes = self.getShapes([1, 2, 10])
215 shapes, dtypes=(dtypes_lib.complex64, dtypes_lib.complex128))
220 shapes = self.getShapes([self._backprop_block_size + 1])
222 shapes,
229 shapes = self.getShapes([2 * self._backprop_block_size + 1])
231 shapes, dtypes=(dtypes_lib.float32,), scalar_test=True)
236 shapes = self.getShapes([2 * self._backprop_block_size + 1])
238 shapes, dtypes=(dtypes_lib.float64,), scalar_test=True)
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_outfeed_ops.cc | 53 std::vector<PartialTensorShape> shapes; in __anondf39fdd70102() local
55 TF_RETURN_IF_ERROR(c->GetAttr("shapes", &shapes)); in __anondf39fdd70102()
57 if (shapes.size() != dtypes.size()) { in __anondf39fdd70102()
61 for (int i = 0; i < shapes.size(); ++i) { in __anondf39fdd70102()
63 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anondf39fdd70102()
87 std::vector<PartialTensorShape> shapes; in __anondf39fdd70202() local
89 TF_RETURN_IF_ERROR(c->GetAttr("shapes", &shapes)); in __anondf39fdd70202()
91 if (shapes.size() != dtypes.size()) { in __anondf39fdd70202()
95 for (int i = 0; i < shapes.size(); ++i) { in __anondf39fdd70202()
97 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anondf39fdd70202()
|
| D | tpu_infeed_ops.cc | 56 std::vector<PartialTensorShape> shapes; in __anon06b801f60102() local
57 TF_RETURN_IF_ERROR(c->GetAttr("shapes", &shapes)); in __anon06b801f60102()
58 for (int i = 0; i < shapes.size(); ++i) { in __anon06b801f60102()
60 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(shapes[i], &out)); in __anon06b801f60102()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 193 shapes=[element_shape],
681 def _shapes(tensor_list_list, shapes, enqueue_many): argument
698 if shapes is None:
706 shapes = [_merge_shapes(
709 return shapes
761 enqueue_many=False, shapes=None, dynamic_pad=False, argument
777 shapes = _shapes([tensor_list], shapes, enqueue_many)
780 capacity=capacity, dtypes=types, shapes=shapes, shared_name=shared_name)
801 enqueue_many=False, shapes=None, dynamic_pad=False, argument
817 shapes = _shapes(tensor_list_list, shapes, enqueue_many)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bcast_ops.cc | 42 absl::InlinedVector<BCast::Vec, 2> shapes; in Compile() local
50 shapes.push_back(BCast::Vec(shape.begin(), shape.end())); in Compile()
52 BCast bcast(shapes[0], shapes[1]); in Compile()
55 "Incompatible shapes: [", absl::StrJoin(shapes[0], ","), in Compile()
56 "] vs. [", absl::StrJoin(shapes[1], ","), "]")); in Compile()
91 absl::InlinedVector<BCast::Vec, 4> shapes; in Compile() local
100 shapes.push_back(BCast::Vec(vec.begin(), vec.end())); in Compile()
102 BCast bcast(shapes[0], shapes[1]); in Compile()
105 "Incompatible shapes: [", absl::StrJoin(shapes[0], ","), in Compile()
106 "] vs. [", absl::StrJoin(shapes[1], ","), "]")); in Compile()
|
| D | pack_op.cc | 47 std::vector<TensorShape> shapes; in Compile() local
48 OP_REQUIRES_OK(ctx, ctx->InputList("values", &values, &shapes)); in Compile()
56 OP_REQUIRES(ctx, shapes[0].IsSameSize(shapes[i]), in Compile()
59 shapes[0].DebugString(), " != values[", i, "].shape = ", in Compile()
60 shapes[i].DebugString())); in Compile()
63 int expanded_num_dims = shapes[0].dims() + 1; in Compile()
74 TensorShape child_shape(shapes[0]); in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | util_test.cc | 49 void SetInputShapes(const std::vector<std::vector<int>>& shapes) { in SetInputShapes() argument
51 tensors_[i].dims = TfLiteIntArrayCreate(shapes[i].size()); in SetInputShapes()
52 std::copy(shapes[i].begin(), shapes[i].end(), tensors_[i].dims->data); in SetInputShapes()
72 std::vector<std::vector<int>> shapes = { in TEST_F() local
75 for (const auto& shape : shapes) { in TEST_F()
84 std::vector<std::vector<int>> shapes = {{4}, {2, 3, 1}, {1, 1, 1, 1}}; in TEST_F() local
85 for (const auto& shape : shapes) { in TEST_F()
102 std::vector<std::vector<int>> shapes = { in TEST_F() local
104 for (const auto& shape : shapes) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bcast_ops.cc | 34 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compute() local
44 shapes.push_back(vec); in Compute()
46 BCast bcast(shapes[0], shapes[1]); in Compute()
49 "Incompatible shapes: [", absl::StrJoin(shapes[0], ","), in Compute()
50 "] vs. [", absl::StrJoin(shapes[1], ","), "]")); in Compute()
83 gtl::InlinedVector<BCast::Vec, 4> shapes; in Compute() local
93 shapes.push_back(vec); in Compute()
95 BCast bcast(shapes[0], shapes[1]); in Compute()
98 "Incompatible shapes: [", absl::StrJoin(shapes[0], ","), in Compute()
99 "] vs. [", absl::StrJoin(shapes[1], ","), "]")); in Compute()
|
| D | sparse_concat_op.cc | 64 OpInputList shapes; in Compute() local
65 OP_REQUIRES_OK(context, context->input_list("shapes", &shapes)); in Compute()
66 OP_REQUIRES(context, shapes.size() == N, in Compute()
68 shapes.size())); in Compute()
70 OP_REQUIRES(context, TensorShapeUtils::IsVector(shapes[i].shape()), in Compute()
73 shapes[i].shape().DebugString(), " at position ", i)); in Compute()
76 const TensorShape input_shape(shapes[0].vec<int64>()); in Compute()
86 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute()
126 const TensorShape current_shape(shapes[i].vec<int64>()); in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | cpu_backend_gemm_test.cc | 537 void TestRandomGemms(const std::vector<std::tuple<int, int, int>>& shapes) { in TestRandomGemms() argument
542 for (const auto& shape : shapes) { in TestRandomGemms()
567 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
569 shapes.push_back(std::make_tuple(size, size, size)); in TYPED_TEST()
571 TestRandomGemms<TypeParam>(shapes); in TYPED_TEST()
575 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
577 shapes.push_back(std::make_tuple(size, size, size)); in TYPED_TEST()
579 TestRandomGemms<TypeParam>(shapes); in TYPED_TEST()
583 std::vector<std::tuple<int, int, int>> shapes; in TYPED_TEST() local
585 shapes.push_back(std::make_tuple(size, size, 1)); in TYPED_TEST()
[all …]
|
| /external/libaom/libaom/test/ |
| D | av1_nn_predict_test.cc | 61 void RunNnPredictTest_all(const NN_CONFIG *const shapes,
63 void RunNnPredictSpeedTest_all(const NN_CONFIG *const shapes,
182 static const NN_CONFIG shapes[] = { variable
192 void NnPredictTest::RunNnPredictTest_all(const NN_CONFIG *const shapes, in RunNnPredictTest_all() argument
194 for (int i = 0; i < num_shapes; i++) RunNnPredictTest(&shapes[i]); in RunNnPredictTest_all()
197 void NnPredictTest::RunNnPredictSpeedTest_all(const NN_CONFIG *const shapes, in RunNnPredictSpeedTest_all() argument
201 NnPredictTest::RunNnPredictSpeedTest(&shapes[i], run_times); in RunNnPredictSpeedTest_all()
205 RunNnPredictTest_all(shapes, sizeof(shapes) / sizeof(*shapes)); in TEST_P()
209 RunNnPredictSpeedTest_all(shapes, sizeof(shapes) / sizeof(*shapes), 10000000); in TEST_P()
|
| /external/harfbuzz_ng/src/ |
| D | gen-arabic-table.py | 153 shapes = {}
186 if items[0] not in shapes:
187 shapes[items[0]] = {}
188 shapes[items[0]][shape] = c
194 keys = shapes.keys ()
197 s = [shapes[u][shape] if u in shapes and shape in shapes[u] else 0
213 liga = (shapes[pair[0]]['initial'], shapes[pair[1]]['final'])
215 liga = (shapes[pair[0]]['medial'], shapes[pair[1]]['final'])
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | base_test.py | 147 shapes = [[2, 32, 32, 3]]
148 return self.BuildParams(self.GraphFn, dtypes.float32, input_shapes=shapes,
149 output_shapes=shapes)
192 shapes = [[2, 32, 32, 3]]
193 return self.BuildParams(self.GraphFn, dtypes.float32, input_shapes=shapes,
194 output_shapes=shapes)
216 shapes = [[2, 32, 32, 3]]
217 return self.BuildParams(self.GraphFn, dtypes.float32, input_shapes=shapes,
218 output_shapes=shapes)
241 shapes = [[2, 32, 32, 3]]
[all …]
|
| /external/kotlinx.coroutines/ui/kotlinx-coroutines-android/animation-app/app/src/main/java/org/jetbrains/kotlinx/animation/ |
| D | Animation.kt | 37 private var shapes = NO_SHAPES variable
41 override fun onChanged(shapes: Set<AnimatedShape>?) { in onChanged()
42 this.shapes = shapes ?: NO_SHAPES in onChanged()
48 shapes.forEach { shape -> in onDraw()
68 private val shapes = MutableLiveData<Set<AnimatedShape>>() constant
71 shapes.observe(owner, observer) in observe()
74 val old = shapes.value ?: NO_SHAPES in observe()
75 shapes.value = if (shape in old) old else old + shape in observe()
86 shapes.value = NO_SHAPES in clearAnimations()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 93 bool IncludesShapes(const std::vector<TensorShape>& shapes) const { in IncludesShapes()
96 if (min.size() != shapes.size()) { in IncludesShapes()
99 for (int i = 0; i < shapes.size(); i++) { in IncludesShapes()
100 auto current_shape = shapes[i]; in IncludesShapes()
141 void AddShape(const std::vector<TensorShape>& shapes) { in AddShape() argument
142 input_shapes_.insert(shapes); in AddShape()
143 VLOG(1) << "Collected shape(s) " << DebugString(shapes) << " for profiles."; in AddShape()
151 int GetProfileNumber(const std::vector<TensorShape>& shapes);
|
| /external/lottie/lottie/src/main/java/com/airbnb/lottie/model/ |
| D | FontCharacter.java | 22 private final List<ShapeGroup> shapes; field in FontCharacter
29 public FontCharacter(List<ShapeGroup> shapes, char character, double size, in FontCharacter() argument
31 this.shapes = shapes; in FontCharacter()
40 return shapes; in getShapes()
|
| /external/gemmlowp/test/ |
| D | benchmark_all_sizes.cc | 282 std::vector<Shape> shapes; in all_shapes_in_random_order() local
290 shapes.push_back(shape); in all_shapes_in_random_order()
298 shapes.push_back(shape); in all_shapes_in_random_order()
309 shapes.push_back(shape); in all_shapes_in_random_order()
314 #error What shapes should we benchmark? (Suggestion: #define BENCHMARK_QUICK) in all_shapes_in_random_order()
316 std::shuffle(std::begin(shapes), std::end(shapes), RandomEngine()); in all_shapes_in_random_order()
317 return shapes; in all_shapes_in_random_order()
321 std::vector<Shape> shapes; in run_benchmarks() local
324 shapes.insert(std::end(shapes), std::begin(pass_shapes), in run_benchmarks()
329 for (std::size_t i = 0; i < shapes.size(); i++) { in run_benchmarks()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_PrelinearizeTuple.pbtxt | 17 name: "shapes"
19 The shapes of each tensor in `inputs`.
26 tuple shapes in the order the shapes appear in the "shapes" input. The layout
|
| D | api_def_InfeedEnqueueTuple.pbtxt | 17 name: "shapes"
19 The shapes of each tensor in `inputs`.
26 all the tuple shapes, in the order the shapes appear in the "shapes" input.
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 80 std::vector<TensorShape>* shapes) { in DynamicShapesToTensorShapes() argument
81 shapes->resize(dynamic_shapes.size()); in DynamicShapesToTensorShapes()
84 ShapeTensorToTensorShape(dynamic_shapes[i], &(*shapes)[i])); in DynamicShapesToTensorShapes()
90 std::vector<TensorShape>* shapes) { in DynamicShapesToTensorShapes() argument
91 shapes->resize(dynamic_shapes.end() - dynamic_shapes.begin()); in DynamicShapesToTensorShapes()
95 ShapeTensorToTensorShape(dynamic_shape.tensor(), &(*shapes)[i])); in DynamicShapesToTensorShapes()
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | host_compute_ops.cc | 66 const AttrValue* shapes; in __anon8acef0f50202() local
67 TF_RETURN_IF_ERROR(c->attrs().Find("shapes", &shapes)); in __anon8acef0f50202()
68 if (shapes->list().shape_size() != c->num_outputs()) { in __anon8acef0f50202()
71 " outputs but 'shapes' attr has ", shapes->list().shape_size(), in __anon8acef0f50202()
77 c->MakeShapeFromShapeProto(shapes->list().shape(i), &handle)); in __anon8acef0f50202()
|
| /external/lottie/lottie/src/main/java/com/airbnb/lottie/model/layer/ |
| D | Layer.java | 36 private final List<ContentModel> shapes; field in Layer
59 …public Layer(List<ContentModel> shapes, LottieComposition composition, String layerName, long laye… in Layer() argument
66 this.shapes = shapes; in Layer()
143 return shapes; in getShapes()
202 if (!shapes.isEmpty()) { in toString()
204 for (Object shape : shapes) { in toString()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | test_util.cc | 30 std::vector<PartialTensorShape> shapes; in ShapeAnnotationsMatch() local
31 for (const auto& output : sit->second) shapes.push_back(output.shape); in ShapeAnnotationsMatch()
35 if (!PartialTensorShapeUtils::AreIdentical(shapes, it->second)) { in ShapeAnnotationsMatch()
40 PartialTensorShapeUtils::PartialShapeListString(shapes)); in ShapeAnnotationsMatch()
|