| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 364 const int64_t d3 = values.begin()->begin()->begin()->size(); in CreateR4WithLayout() local
521 const Shape& shape, in CreateLiteralWithGenerator()
533 const Shape& shape, E* engine, T mean, T stddev) { in CreateRandomLiteral()
544 const Shape& shape, T mean, T stddev) { in CreateRandomLiteral()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 141 TensorShape shape = {}; in TEST_F() local
165 Device* d3 = ListDevices().at(3); in TEST_F() local
220 TensorShape shape = {}; in TEST_F() local
261 TensorShape shape = {2}; in TEST() local
388 TensorShape shape = {}; in TEST_F() local
423 TensorShape shape = {2}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 424 std::unique_ptr<TfLiteIntArray, void (*)(TfLiteIntArray*)> shape( in GetOutputShapeFromInput() local
438 const std::string GetShapeDebugString(const TfLiteIntArray* shape) { in GetShapeDebugString()
464 std::unique_ptr<TfLiteIntArray, void (*)(TfLiteIntArray*)> shape( in CalculateShapeForBroadcast() local
496 std::unique_ptr<TfLiteIntArray, void (*)(TfLiteIntArray*)> shape( in CalculateShapeForBroadcast() local
501 const int d3 = i >= dims3 ? 1 : SizeOfDimension(input3, dims3 - i - 1); in CalculateShapeForBroadcast() local
|
| /external/robolectric/integration_tests/ctesque/src/sharedTest/java/android/graphics/drawable/ |
| D | GradientDrawableTest.java | 358 int shape; in testSetShape() local
782 GradientDrawable d3 = (GradientDrawable) mResources.getDrawable(R.drawable.gradientdrawable); in testMutate() local
|
| /external/pytorch/test/ |
| D | test_sparse_csr.py | 468 def run_test(shape, blocksize, nnz, index_type): argument
649 def shape(shape, basedim=0): argument
1350 def get_dense_shape(shape, batch_ndim): argument
1941 def test_shape(d1, d2, d3, nnz, transposed, index_dtype): argument
2163 def _test_spadd_shape(nnz, shape): argument
2208 def _test_spadd_shape(fn, nnz, shape): argument
2966 def run_test(shape, nnz, index_type): argument
3033 def run_test(shape, nnz, index_type, n_dense, blocksize=()): argument
|
| D | test_sparse.py | 442 def func(indices, values, shape, is_coalesced): argument
1599 def test_shape(d1, d2, d3, nnz, transposed): argument
2170 def _test_zeros(nnzs, shape, out_shape_i, out_shape_v=None): argument
2181 def test_shape(i_shapes, v_shapes, shape, nnzs): argument
2325 def _all_narrow_combs(self, shape): argument
3928 def reference(diags, offsets, shape): argument
3932 def reference(diags, offsets, shape): argument
3944 def check_valid(diags, offsets, shape, layout=None): argument
4019 def run_test(shape, nnz): argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 1188 Shape shape = ShapeUtil::MakeShape(F32, {2, 2}); in XLA_TEST_F() local
1829 const int64_t d3 = 128; in DOT_ReorderContracting() local
|
| D | array_elementwise_ops_test.cc | 3074 constexpr int d3 = 2; in XLA_TEST_F() local
3102 auto shape = ShapeUtil::MakeOpaqueShape(); in XLA_TEST_F() local
|