| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | softmax_spmd_expander.cc | 372 mlir::Operation* op, bool is_sparse, const Layout& features_layout, in MaybeRelayoutInputs()
525 const bool is_sparse = in ExpandOp() local
625 const bool is_sparse = in ComputeLayoutForward() local
678 const bool is_sparse = in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/python/tpu/tests/ |
| D | tpu_embedding_v2_sequence_feature_test.py | 29 def test_sequence_feature(self, is_sparse): argument
|
| D | tpu_embedding_v2_valid_input_test.py | 247 def test_missing_feature(self, is_sparse): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | proximal_gradient_descent_test.py | 135 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
|
| D | proximal_adagrad_test.py | 172 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
|
| D | ftrl_test.py | 371 def applyOptimizer(self, opt, dtype, steps=5, is_sparse=False): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 35 explicit MatMulOp(OpKernelConstruction* ctx, bool is_sparse = false) in MatMulOp()
|
| /external/tensorflow/tensorflow/python/ |
| D | pywrap_dtensor_device.cc | 211 const py::capsule& device_info, const bool is_sparse) { in PYBIND11_MODULE()
310 bool is_sparse = IsSparseDTensor( in PYBIND11_MODULE() local
|
| /external/cronet/net/disk_cache/ |
| D | disk_cache_fuzzer.proto | 92 required bool is_sparse = 5; field
108 required bool is_sparse = 6; field
|
| D | disk_cache_fuzzer.cc | 494 bool is_sparse = ce.is_sparse(); in RunCommands() local
578 bool is_sparse = ooce.is_sparse(); in RunCommands() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_sparsetensors.cc | 140 const auto is_sparse = main_func.getArgAttrOfType<mlir::BoolAttr>( in CreateComponentTensorsFromSparseTensors() local
|
| D | collectives.cc | 218 bool is_sparse = IsSparseValue(input); in EmitRelayout() local
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 393 def is_sparse(tensor): function
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor.py | 468 def is_sparse(x): function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 338 const bool is_sparse = filter->sparsity != nullptr; in PrepareImpl() local
|
| D | lstm.cc | 1938 const bool is_sparse = input_to_output_weights->sparsity != nullptr; in Eval() local
|
| /external/libaom/av1/encoder/ |
| D | optical_flow.c | 43 static INLINE int is_sparse(const OPFL_PARAMS *opfl_params) { in is_sparse() function
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1561 bool is_sparse = d_and_type.second == Type::Sparse; in FastParseSingleExample() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 960 def is_sparse(tensor): function
|