/external/tensorflow/tensorflow/core/kernels/ |
string_split_op.cc | 107 auto sp_shape = sp_shape_t->vec<int64>(); variable 108 sp_shape(0) = batch_size; 109 sp_shape(1) = max_num_entries;
|
sparse_dense_binary_op_shared_test.cc | 212 Node* sp_shape, Node* dense) { 218 .Input(sp_shape)
|
sparse_reduce_op.cc | 85 auto sp_shape = sp.shape(); local 95 out_dim_sizes.push_back(sp_shape[d]);
|
/external/tensorflow/tensorflow/python/ops/ |
sparse_grad.py | 112 sp_shape = op.inputs[2] 113 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3]) 115 scale = sp_shape // math_ops.to_int64(output_shape_kept_dims) 213 # (sp_indices, sp_vals, sp_shape, dense) 244 Gradients w.r.t. the input (sp_indices, sp_values, sp_shape).
|
array_grad.py | 786 sp_shape = (rows_in * cols_in, rows_out * cols_out * ksize_r * ksize_c) 790 array_ops.ones((len(idx),), dtype=ops.dtypes.float32), sp_shape)
|
math_ops.py | 916 sp_values, sp_shape, dense)` and outputs `(new_sp_values)`. 995 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): 998 [sp_indices, sp_values, sp_shape, y]) as name: 1014 sp_indices, sp_values, sp_shape, y, name=name) [all...] |
/external/tensorflow/tensorflow/python/client/ |
session_test.py | 616 sp_shape = array_ops.identity(sp.dense_shape) 617 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape) 620 [sp_indices, sp_values, sp_shape], { 633 [sp_indices, sp_values, sp_shape], { 662 sp_shape = array_ops.identity(sp.dense_shape) 663 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape) 666 [sp_indices, sp_values, sp_shape], { 674 [sp_indices, sp_values, sp_shape], { 697 sp_shape = array_ops.identity(sp.dense_shape) 698 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape) [all...] |
/external/tensorflow/tensorflow/python/kernel_tests/ |
sparse_ops_test.py | 730 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes): 731 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1 772 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes): 773 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1 [all...] |
lookup_ops_test.py | 162 sp_shape = [2, 2] 166 constant_op.constant(sp_shape, dtypes.int64)) 173 self.assertAllEqual(sp_shape, out_shape) [all...] |
/external/tensorflow/tensorflow/contrib/lookup/ |
lookup_ops_test.py | 168 sp_shape = [2, 2] 172 constant_op.constant(sp_shape, dtypes.int64)) 179 self.assertAllEqual(sp_shape, out_shape) [all...] |
/external/tensorflow/tensorflow/go/op/ |
wrappers.go | [all...] |