Searched defs:b_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 42 const TensorShape b_shape = ctx->InputShape(1); in Compile() local
|
| D | matmul_op.cc | 54 const TensorShape b_shape = ctx->InputShape(1); 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/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/contrib/gan/python/features/python/ |
| D | spectral_normalization_test.py | 107 def _testLayerHelper(self, build_layer_fn, w_shape, b_shape, is_keras=False): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 129 ShapeHandle b_shape; in __anonb22bfa860202() local
1460 ShapeHandle b_shape; in __anonb22bfa860c02() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 556 const Shape& b_shape = hlo->operand(1)->shape(); in HandleTriangularSolve() local
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 66 Eigen::array<Eigen::DenseIndex, 2> b_shape({1, b.dimensions()[0]}); in LSTMBlockCellFpropWithEigen() local
|