| /external/tensorflow/tensorflow/core/ops/ |
| D | training_ops.cc | 51 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/dtensor/mlir/expansions/ |
| D | softmax_spmd_expander.cc | 372 mlir::Operation* op, bool is_sparse, const Layout& features_layout, in MaybeRelayoutInputs() argument
394 (is_sparse && labels_layout.rank() == 1)) && in MaybeRelayoutInputs()
410 if (!is_sparse && labels_layout.rank() > 0 && in MaybeRelayoutInputs()
442 if (is_sparse) { in MaybeRelayoutInputs()
525 const bool is_sparse = in ExpandOp() local
533 MaybeRelayoutInputs(op, is_sparse, features_layout, labels_layout, in ExpandOp()
552 if (is_sparse) { in ExpandOp()
625 const bool is_sparse = in ComputeLayoutForward() local
651 (is_sparse && labels_layout->rank() == 1)) && in ComputeLayoutForward()
661 if (!is_sparse && labels_layout && (labels_layout->rank() > 0) && in ComputeLayoutForward()
[all …]
|
| D | softmax_spmd_expander.h | 57 StatusOr<Layout> MaybeRelayoutInputs(mlir::Operation* op, bool is_sparse,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 35 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/ |
| D | proximal_gradient_descent_test.py | 135 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
136 if is_sparse:
159 if is_sparse:
180 is_sparse=True)
183 gradient_descent.GradientDescentOptimizer(3.0), is_sparse=True)
|
| D | proximal_adagrad_test.py | 172 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
173 if is_sparse:
197 if is_sparse:
238 is_sparse=True)
244 is_sparse=True)
|
| D | ftrl_test.py | 371 def applyOptimizer(self, opt, dtype, steps=5, is_sparse=False): argument
372 if is_sparse:
392 if is_sparse:
450 is_sparse=True)
456 is_sparse=True)
475 is_sparse=True)
481 is_sparse=True)
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | dtensor_device.py | 225 is_sparse = True
232 is_sparse = False
239 is_sparse)
277 is_sparse = _pywrap_dtensor_device.IsSparseDTensor(
281 if is_sparse:
|
| /external/tensorflow/tensorflow/python/tpu/tests/ |
| D | tpu_embedding_v2_sequence_feature_test.py | 29 def test_sequence_feature(self, is_sparse): argument
35 if is_sparse:
|
| D | tpu_embedding_v2_valid_input_test.py | 247 def test_missing_feature(self, is_sparse): argument
256 if is_sparse:
|
| /external/cronet/net/disk_cache/ |
| D | disk_cache_fuzzer.cc | 494 bool is_sparse = ce.is_sparse(); in RunCommands() local
507 base::Unretained(this), entry_id, async, is_sparse); in RunCommands()
510 << "\", set_is_sparse = " << is_sparse in RunCommands()
521 sparse_entry_tracker_[entry_info->entry_ptr] = is_sparse; in RunCommands()
578 bool is_sparse = ooce.is_sparse(); in RunCommands() local
604 base::Unretained(this), entry_id, async, is_sparse); in RunCommands()
608 << "\", set_is_sparse = " << is_sparse in RunCommands()
619 sparse_entry_tracker_[entry_info->entry_ptr] = is_sparse; in RunCommands()
|
| D | disk_cache_fuzzer.proto | 92 required bool is_sparse = 5; field
108 required bool is_sparse = 6; field
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor_test.py | 76 self.assertFalse(sparse_tensor.is_sparse(3))
77 self.assertFalse(sparse_tensor.is_sparse("foo"))
78 self.assertFalse(sparse_tensor.is_sparse(np.array(3)))
80 sparse_tensor.is_sparse(sparse_tensor.SparseTensor([[0]], [0], [1])))
82 sparse_tensor.is_sparse(
|
| D | sparse_tensor.py | 104 if not is_sparse(sparse_tensor_value):
468 def is_sparse(x): function
|
| /external/tensorflow/tensorflow/python/ |
| D | pywrap_dtensor_device.cc | 211 const py::capsule& device_info, const bool is_sparse) { in PYBIND11_MODULE() argument
233 if (is_sparse) { in PYBIND11_MODULE()
310 bool is_sparse = IsSparseDTensor( in PYBIND11_MODULE() local
318 return is_sparse; in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_sparsetensors.cc | 140 const auto is_sparse = main_func.getArgAttrOfType<mlir::BoolAttr>( in CreateComponentTensorsFromSparseTensors() local
142 if (is_sparse) { in CreateComponentTensorsFromSparseTensors()
|
| D | collectives.cc | 218 bool is_sparse = IsSparseValue(input); in EmitRelayout() local
278 if (!is_sparse) return all_scatter; in EmitRelayout()
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | sparse.py | 127 if sparse_tensor.is_sparse(tensor) else tensor
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | test_base.py | 85 elif sparse_tensor.is_sparse(expected):
262 if sparse_tensor.is_sparse(op1[i]) or ragged_tensor.is_ragged(op1[i]):
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 1756 is_sparse = {}
1757 is_sparse[op.inputs[0].ref()] = op.get_attr("a_is_sparse")
1758 is_sparse[op.inputs[1].ref()] = op.get_attr("b_is_sparse")
1760 is_sparse[grad.ref()] = not context.executing_eagerly() and (
1766 assert t1.ref() in is_sparse and t2.ref() in is_sparse
1767 t1_sparse = is_sparse[t1.ref()]
1768 t2_sparse = is_sparse[t2.ref()]
|
| /external/avb/ |
| D | avbtool.py | 771 self.is_sparse = False
881 self.is_sparse = True
912 if not self.is_sparse:
953 if not self.is_sparse:
992 if not self.is_sparse:
1039 if not self.is_sparse:
1096 if not self.is_sparse:
2444 ' (Sparse)' if image.is_sparse else ''))
3297 assert not image.is_sparse
3512 assert not image.is_sparse
[all …]
|
| /external/libaom/av1/encoder/ |
| D | optical_flow.c | 43 static INLINE int is_sparse(const OPFL_PARAMS *opfl_params) { in is_sparse() function
1011 if (is_sparse(opfl_params)) { in pyramid_optical_flow()
1021 assert(is_sparse(opfl_params)); in pyramid_optical_flow()
1026 assert(!is_sparse(opfl_params)); in pyramid_optical_flow()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 393 def is_sparse(tensor): function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_cross_op_test.py | 418 elif sparse_tensor.is_sparse(t):
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 338 const bool is_sparse = filter->sparsity != nullptr; in PrepareImpl() local
342 if (is_sparse) { in PrepareImpl()
416 if (is_sparse) { in PrepareImpl()
|