Home
last modified time | relevance | path

Searched refs:is_sparse (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Dtraining_ops.cc51 template <bool is_sparse, bool is_resource>
55 if (!is_sparse) { in HandleGradAndIndicesInputs()
102 template <bool is_sparse, bool is_resource>
109 TF_RETURN_IF_ERROR(HandleGradAndIndicesInputs<is_sparse, is_resource>( in ApplyProximalGradientDescentShapeFn()
167 template <bool is_sparse, bool is_resource>
178 TF_RETURN_IF_ERROR(HandleGradAndIndicesInputs<is_sparse, is_resource>( in ApplyAdadeltaShapeFn()
243 template <bool is_sparse, bool is_resource>
250 TF_RETURN_IF_ERROR(HandleGradAndIndicesInputs<is_sparse, is_resource>( in ApplyAdagradShapeFn()
305 template <bool is_sparse, bool is_resource>
313 TF_RETURN_IF_ERROR(HandleGradAndIndicesInputs<is_sparse, is_resource>( in ApplyAdagradV2ShapeFn()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatmul_op.cc35 explicit MatMulOp(OpKernelConstruction* ctx, bool is_sparse = false) in MatMulOp() argument
36 : XlaOpKernel(ctx), is_sparse_(is_sparse) { in MatMulOp()
39 if (is_sparse) { in MatMulOp()
/external/tensorflow/tensorflow/python/training/
Dproximal_gradient_descent_test.py138 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
139 if is_sparse:
162 if is_sparse:
183 is_sparse=True)
186 gradient_descent.GradientDescentOptimizer(3.0), is_sparse=True)
Dproximal_adagrad_test.py175 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
176 if is_sparse:
200 if is_sparse:
241 is_sparse=True)
247 is_sparse=True)
Dftrl_test.py374 def applyOptimizer(self, opt, dtype, steps=5, is_sparse=False): argument
375 if is_sparse:
395 if is_sparse:
453 is_sparse=True)
459 is_sparse=True)
478 is_sparse=True)
484 is_sparse=True)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dftrl_test.py363 def applyOptimizer(self, opt, dtype, steps=5, is_sparse=False): argument
364 if is_sparse:
383 if is_sparse:
438 is_sparse=True)
444 is_sparse=True)
462 is_sparse=True)
468 is_sparse=True)
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py78 self.assertFalse(sparse_tensor.is_sparse(3))
79 self.assertFalse(sparse_tensor.is_sparse("foo"))
80 self.assertFalse(sparse_tensor.is_sparse(np.array(3)))
82 sparse_tensor.is_sparse(sparse_tensor.SparseTensor([[0]], [0], [1])))
84 sparse_tensor.is_sparse(
Dsparse_tensor.py108 if not is_sparse(sparse_tensor_value):
469 def is_sparse(x): function
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc261 const bool is_sparse = filter->sparsity != nullptr; in PrepareImpl() local
265 if (is_sparse) { in PrepareImpl()
339 if (is_sparse) { in PrepareImpl()
457 const bool is_sparse = filter->sparsity != nullptr; in EvalHybridImpl() local
505 if (is_sparse) { in EvalHybridImpl()
590 const bool is_sparse = filter->sparsity != nullptr; in EvalHybrid() local
591 if (is_sparse) { in EvalHybrid()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dtest_base.py82 elif sparse_tensor.is_sparse(expected):
249 if sparse_tensor.is_sparse(op1[i]) or ragged_tensor.is_ragged(op1[i]):
Dfrom_tensors_test.py110 if sparse_tensor.is_sparse(c) else c.shape for c in components
Dfrom_tensor_slices_test.py152 if sparse_tensor.is_sparse(c) else c.shape[1:] for c in components
/external/tensorflow/tensorflow/python/data/util/
Dsparse.py131 if sparse_tensor.is_sparse(tensor) else tensor
/external/tensorflow/tensorflow/python/ops/
Dmath_grad.py1754 is_sparse = {}
1755 is_sparse[op.inputs[0].ref()] = op.get_attr("a_is_sparse")
1756 is_sparse[op.inputs[1].ref()] = op.get_attr("b_is_sparse")
1758 is_sparse[grad.ref()] = not context.executing_eagerly() and (
1764 assert t1.ref() in is_sparse and t2.ref() in is_sparse
1765 t1_sparse = is_sparse[t1.ref()]
1766 t2_sparse = is_sparse[t2.ref()]
/external/avb/
Davbtool.py770 self.is_sparse = False
880 self.is_sparse = True
911 if not self.is_sparse:
948 if not self.is_sparse:
987 if not self.is_sparse:
1034 if not self.is_sparse:
1091 if not self.is_sparse:
2435 ' (Sparse)' if image.is_sparse else ''))
3279 assert not image.is_sparse
3480 assert not image.is_sparse
[all …]
Davbtool770 self.is_sparse = False
880 self.is_sparse = True
911 if not self.is_sparse:
948 if not self.is_sparse:
987 if not self.is_sparse:
1034 if not self.is_sparse:
1091 if not self.is_sparse:
2435 ' (Sparse)' if image.is_sparse else ''))
3279 assert not image.is_sparse
3480 assert not image.is_sparse
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_cross_op_test.py390 elif sparse_tensor.is_sparse(t):
Dragged_array_ops.py792 if sparse_tensor.is_sparse(t):
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1545 bool is_sparse = d_and_type.second == Type::Sparse; in FastParseSingleExample() local
1552 : (is_sparse ? config.sparse[d].feature_name in FastParseSingleExample()
1651 auto& feature_already_seen = is_sparse ? sparse_feature_already_seen in FastParseSingleExample()
1654 is_sparse ? config.sparse[d].dtype : config.ragged[d].dtype; in FastParseSingleExample()
1719 } else if (is_sparse) { in FastParseSingleExample()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_arrays_v1.py234 if issparse(input_data) and not K.is_sparse(feed_tensor):
Dtraining_v1.py2996 sparse=K.is_sparse(self.output),
3163 if K.is_sparse(expected_input):
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_test.py1098 map(sparse_tensor.is_sparse, per_replica_values.values)) else
1241 map(sparse_tensor.is_sparse, per_replica_values.values)) else
/external/tensorflow/tensorflow/python/keras/
Dbackend.py963 def is_sparse(tensor): function
1012 if is_sparse(tensor):
2010 if is_sparse(x):
3098 if py_all(is_sparse(x) for x in tensors):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.backend.pbtxt248 name: "is_sparse"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.backend.pbtxt256 name: "is_sparse"

12