Home
last modified time | relevance | path

Searched refs:sparse_handles (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_tensors_map_ops_test.py87 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)
137 sparse_map_op=handles.op, sparse_handles=handles)
165 sparse_map_op=handle.op, sparse_handles=handle_concat)
180 sparse_map_op=handle.op, sparse_handles=[handle_value, bad_handle])
208 sparse_map_op=st_handles.op, sparse_handles=st_handles)
/external/tensorflow/tensorflow/core/ops/
Dsdca_ops.cc27 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()
Dsparse_ops.cc592 ShapeHandle sparse_handles; in __anonc55240e61702() local
593 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &sparse_handles)); in __anonc55240e61702()
Dops.pbtxt304 name: "sparse_handles"
57982 name: "sparse_handles"
/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensors_map_ops.cc279 Tensor sparse_handles(DT_INT64, TensorShape({N})); in Compute() local
280 auto sparse_handles_t = sparse_handles.vec<int64_t>(); in Compute()
349 context->set_output(0, sparse_handles); in Compute()
372 const Tensor& sparse_handles = context->input(0); in Compute() local
374 OP_REQUIRES(context, TensorShapeUtils::IsVector(sparse_handles.shape()), in Compute()
377 sparse_handles.shape().DebugString())); in Compute()
379 int64_t N = sparse_handles.shape().dim_size(0); in Compute()
390 const auto& sparse_handles_t = sparse_handles.vec<int64_t>(); in Compute()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AddManySparseToTensorsMap.pbtxt24 name: "sparse_handles"
57 `SparseTensor` objects pointed to by each row of the output `sparse_handles`
Dapi_def_TakeManySparseFromTensorsMap.pbtxt4 name: "sparse_handles"
51 The input `sparse_handles` must be an `int64` matrix of shape `[N, 1]` where
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DAddManySparseToTensorsMap.pbtxt16 name: "sparse_handles"
DTakeManySparseFromTensorsMap.pbtxt4 name: "sparse_handles"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DAddManySparseToTensorsMap.pbtxt16 name: "sparse_handles"
DTakeManySparseFromTensorsMap.pbtxt4 name: "sparse_handles"
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py3047 sparse_handles, argument
3119 sparse_handles,
/external/tensorflow/tensorflow/python/training/
Dinput.py602 sparse_handles=array_ops.squeeze(s, [1]),
/external/tensorflow/tensorflow/go/op/
Dwrappers.go262 …ut, sparse_shape tf.Output, optional ...AddManySparseToTensorsMapAttr) (sparse_handles tf.Output) {
50721 func TakeManySparseFromTensorsMap(scope *Scope, sparse_handles tf.Output, dtype tf.DataType, option…
50732 sparse_handles,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4913 …argspec: "args=[\'sparse_handles\', \'dtype\', \'container\', \'shared_name\', \'name\'], varargs=…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4913 …argspec: "args=[\'sparse_handles\', \'dtype\', \'container\', \'shared_name\', \'name\'], varargs=…
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt378 name: "sparse_handles"
36611 name: "sparse_handles"
Dops_history.v2.pbtxt855 name: "sparse_handles"
81644 name: "sparse_handles"