| /external/tensorflow/tensorflow/core/kernels/ |
| string_split_op.cc | 224 auto sp_shape = sp_shape_t->vec<int64>(); variable
225 sp_shape(0) = batch_size;
226 sp_shape(1) = max_num_entries;
294 auto sp_shape = sp_shape_t->vec<int64>(); variable
295 sp_shape(0) = batch_size;
296 sp_shape(1) = max_num_entries;
|
| sparse_dense_binary_op_shared_test.cc | 213 Node* sp_shape, Node* dense) {
219 .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 | 110 sp_shape = op.inputs[2]
111 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
113 scale = sp_shape // math_ops.cast(output_shape_kept_dims, dtypes.int64)
240 # (sp_indices, sp_vals, sp_shape, dense)
271 Gradients w.r.t. the input (sp_indices, sp_values, sp_shape).
|
| array_grad.py | 776 sp_shape = (input_indices_num, output_indices_num)
780 sp_shape)
840 sp_shape = (input_indices_num, output_indices_num)
842 idx_map, array_ops.ones([output_indices_num], dtype=grad.dtype), sp_shape)
[all...] |
| math_ops.py | 852 sp_values, sp_shape, dense)` and outputs `(new_sp_values)`.
934 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None):
937 [sp_indices, sp_values, sp_shape, y]) as name:
953 sp_indices, sp_values, sp_shape, y, name=name)
[all...] |
| /external/tensorflow/tensorflow/python/client/ |
| session_test.py | 701 sp_shape = array_ops.identity(sp.dense_shape)
702 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
705 [sp_indices, sp_values, sp_shape], {
718 [sp_indices, sp_values, sp_shape], {
747 sp_shape = array_ops.identity(sp.dense_shape)
748 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
751 [sp_indices, sp_values, sp_shape], {
759 [sp_indices, sp_values, sp_shape], {
782 sp_shape = array_ops.identity(sp.dense_shape)
783 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_ops_test.py | [all...] |
| lookup_ops_test.py | 192 sp_shape = [2, 2]
196 constant_op.constant(sp_shape, dtypes.int64))
203 self.assertAllEqual(sp_shape, out_shape)
[all...] |
| /external/tensorflow/tensorflow/contrib/lookup/ |
| lookup_ops_test.py | 172 sp_shape = [2, 2]
176 constant_op.constant(sp_shape, dtypes.int64))
183 self.assertAllEqual(sp_shape, out_shape)
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
