| /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 | 44 const Tensor* a_shape; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 41 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/contrib/factorization/python/kernel_tests/ |
| D | masked_matmul_benchmark.py | 62 def _run_graph(self, a_shape, b_shape, nnz, num_iters, sort=False, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 109 def CompareNonEmpty(self, a_shape, b_shape): argument
126 def CompareEmpty(self, a_shape, b_shape): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 76 ShapeHandle a_shape; in __anone6e195410302() local
99 ShapeHandle a_shape; in __anone6e195410402() local
|
| D | math_ops.cc | 128 ShapeHandle a_shape; in __anon57f5a5dd0202() local
1459 ShapeHandle a_shape; in __anon57f5a5dd0c02() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 57 Shape a_shape = operand->shape(); in CreateCholesky() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 555 const Shape& a_shape = hlo->operand(0)->shape(); in HandleTriangularSolve() local
568 const Shape& a_shape = hlo->operand(0)->shape(); in HandleCholesky() local
|
| D | shape_inference_test.cc | 1521 Shape a_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5}); in TEST_F() local
1531 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 | 835 Shape a_shape = shape_with_status.ValueOrDie(); in SVD() local
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 189 const TensorShape& a_shape = found_shapes[node.input(0)]; in CalculateFlops() local
|