| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 80 def CompareNonEmpty(self, a_shape, b_shape): argument
100 def CompareNonEmpty(self, a_shape, b_shape): argument
118 def CompareEmpty(self, a_shape, b_shape): argument
176 def _compare(self, a_shape, b_shape, dtype, adjoint_a, adjoint_b): argument
186 def CheckGradients(self, a_shape, b_shape): argument
198 def CheckGradients(self, a_shape, b_shape): argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | shared_device_buffer_test.cc | 45 Shape a_shape = ShapeUtil::MakeShape(F32, {3, 101, 4}); in TEST() local
73 Shape a_shape = ShapeUtil::MakeShape(F32, {3, 101, 4}); in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op_test.cc | 27 Node* a_shape, Node* b, bool adjoint_a, in SparseTensorDenseMatMulNode()
47 Tensor a_shape(DT_INT64, TensorShape({2})); in SparseTensorDenseMatmul() local
|
| D | betainc_op.cc | 47 const TensorShape& a_shape = a.shape(); in Compute() local
|
| D | sparse_tensor_dense_add_op.cc | 36 const Tensor *a_shape, const Tensor *b) { in ValidateInputs()
|
| D | sparse_add_op.cc | 32 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
|
| D | sparse_sparse_binary_op_shared.cc | 174 const auto a_shape = a_shape_t->flat<int64>(); in Compute() local
|
| D | sparse_tensor_dense_matmul_op.cc | 45 const Tensor* a_shape; in Compute() local
|
| /external/XNNPACK/models/ |
| D | mobilenet-v3-small.cc | 2163 const size_t a_shape[] = { 1, 56, 56, 16 }; in MobileNetV3Small() local
2327 const size_t a_shape[] = { 1, 14, 14, 96 }; in MobileNetV3Small() local
2421 const size_t a_shape[] = { 1, 14, 14, 240 }; in MobileNetV3Small() local
2525 const size_t a_shape[] = { 1, 14, 14, 240 }; in MobileNetV3Small() local
2629 const size_t a_shape[] = { 1, 14, 14, 120 }; in MobileNetV3Small() local
2723 const size_t a_shape[] = { 1, 14, 14, 144 }; in MobileNetV3Small() local
2827 const size_t a_shape[] = { 1, 7, 7, 288 }; in MobileNetV3Small() local
2921 const size_t a_shape[] = { 1, 7, 7, 576 }; in MobileNetV3Small() local
3025 const size_t a_shape[] = { 1, 7, 7, 576 }; in MobileNetV3Small() local
|
| D | mobilenet-v3-large.cc | 2610 const size_t a_shape[] = { 1, 28, 28, 72 }; in MobileNetV3Large() local
2684 const size_t a_shape[] = { 1, 28, 28, 120 }; in MobileNetV3Large() local
2768 const size_t a_shape[] = { 1, 28, 28, 120 }; in MobileNetV3Large() local
3102 const size_t a_shape[] = { 1, 14, 14, 480 }; in MobileNetV3Large() local
3196 const size_t a_shape[] = { 1, 14, 14, 672 }; in MobileNetV3Large() local
3300 const size_t a_shape[] = { 1, 7, 7, 672 }; in MobileNetV3Large() local
3394 const size_t a_shape[] = { 1, 7, 7, 960 }; in MobileNetV3Large() local
3498 const size_t a_shape[] = { 1, 7, 7, 960 }; in MobileNetV3Large() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops_test.cc | 192 const string& a_shape, const string& b_shape) { in TEST()
228 const string& a_shape, const string& b_shape) { in TEST()
328 const string& a_shape, const string& b_shape) { in TEST()
|
| D | sparse_csr_matrix_ops.cc | 235 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anond96a6a670702() local
306 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anond96a6a670802() local
359 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anond96a6a670902() local
393 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anond96a6a670a02() local
|
| D | sparse_ops.cc | 76 ShapeHandle a_shape; in __anon5983aeac0302() local
99 ShapeHandle a_shape; in __anon5983aeac0402() local
|
| D | math_ops.cc | 1552 ShapeHandle a_shape; in __anonb5dc68880c02() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | beta_op.cc | 37 const TensorShape& a_shape = ctx->InputShape(0); in Compile() local
|
| D | matmul_op.cc | 53 const TensorShape a_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 55 Shape a_shape = operand->shape(); in CreateCholesky() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 682 const Shape& a_shape = hlo->operand(0)->shape(); in HandleTriangularSolve() local
699 const Shape& a_shape = hlo->operand(0)->shape(); in HandleCholesky() local
|
| D | shape_inference_test.cc | 1790 Shape a_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
1800 Shape a_shape = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | self_adjoint_eig.cc | 406 Shape a_shape = shape_with_status.ValueOrDie(); in SelfAdjointEig() local
|
| D | svd.cc | 822 Shape a_shape = shape_with_status.ValueOrDie(); in SVD() local
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_mat_mul_op.cc | 128 TensorShape a_shape; in Compute() local
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 190 const TensorShape& a_shape = found_shapes[node.input(0)]; in CalculateFlops() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 419 ShapeHandle a_shape; in BatchMatMulV2Shape() local
456 ShapeHandle a_shape; in BatchMatMulShape() local
|