Searched defs:b_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 108 def compareNonEmpty(self, a_shape, b_shape): argument
125 def compareEmpty(self, a_shape, b_shape): argument
|
| /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/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 53 const TensorShape b_shape = ctx->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | betainc_op.cc | 48 const TensorShape& b_shape = b.shape(); in Compute() local
|
| D | sparse_add_op.cc | 33 *b_shape, *thresh_t; in Compute() local
|
| D | sparse_sparse_binary_op_shared.cc | 175 const auto b_shape = b_shape_t->flat<int64>(); in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 73 ShapeHandle b_shape; in __anon57f5a5dd0202() local
1327 ShapeHandle b_shape; in __anon57f5a5dd0a02() local
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 65 Eigen::array<Eigen::DenseIndex, 2> b_shape({1, b.dimensions()[0]}); in LSTMBlockCellFpropWithEigen() local
|