Searched refs:sparse_handles (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensors_map_ops_test.py | 88 sparse_map_op=handle0.op, sparse_handles=handles_concat)
110 sparse_handles = ops.convert_to_tensor(
114 sparse_map_op=handle.op, sparse_handles=sparse_handles)
135 sparse_map_op=handles.op, sparse_handles=handles)
162 sparse_map_op=handle.op, sparse_handles=handle_concat)
176 sparse_map_op=handle.op, sparse_handles=[handle_value, bad_handle])
204 sparse_map_op=st_handles.op, sparse_handles=st_handles)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sdca_ops.cc | 27 std::vector<ShapeHandle> sparse_handles; in ApplySdcaOptimizerShapeFn() local
28 if (c->input("sparse_weights", &sparse_handles).ok()) { in ApplySdcaOptimizerShapeFn()
30 c->set_output("out_delta_sparse_weights", sparse_handles)); in ApplySdcaOptimizerShapeFn()
|
| D | sparse_ops.cc | 535 ShapeHandle sparse_handles; in __anone6e195411402() local
536 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &sparse_handles)); in __anone6e195411402()
|
| D | ops.pbtxt | 300 name: "sparse_handles"
28799 name: "sparse_handles"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensors_map_ops.cc | 262 Tensor sparse_handles(DT_INT64, TensorShape({N})); in Compute() local
263 auto sparse_handles_t = sparse_handles.vec<int64>(); in Compute()
327 context->set_output(0, sparse_handles); in Compute()
350 const Tensor& sparse_handles = context->input(0); in Compute() local
352 OP_REQUIRES(context, TensorShapeUtils::IsVector(sparse_handles.shape()), in Compute()
355 sparse_handles.shape().DebugString())); in Compute()
357 int64 N = sparse_handles.shape().dim_size(0); in Compute()
368 const auto& sparse_handles_t = sparse_handles.vec<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AddManySparseToTensorsMap.pbtxt | 24 name: "sparse_handles"
57 `SparseTensor` objects pointed to by each row of the output `sparse_handles`
|
| D | api_def_TakeManySparseFromTensorsMap.pbtxt | 4 name: "sparse_handles"
51 The input `sparse_handles` must be an `int64` matrix of shape `[N, 1]` where
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 2099 sparse_handles, argument
2171 sparse_handles,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 582 sparse_handles=array_ops.squeeze(s, [1]),
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sequence_queueing_state_saver.py | 1863 sparse_handles=_flatten_tensor(s), rank=None)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 11042 …ut, sparse_shape tf.Output, optional ...AddManySparseToTensorsMapAttr) (sparse_handles tf.Output) {
13475 func TakeManySparseFromTensorsMap(scope *Scope, sparse_handles tf.Output, dtype tf.DataType, option…
13486 sparse_handles,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 378 name: "sparse_handles"
36611 name: "sparse_handles"
|
| D | ops_history.v1.pbtxt | 743 name: "sparse_handles"
62355 name: "sparse_handles"
|