| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A 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
|
| /aosp_15_r20/external/mesa3d/src/intel/vulkan/ |
| H A D | anv_buffer.c | 57 bool is_sparse, in anv_get_buffer_memory_requirements()
142 const bool is_sparse = in anv_GetDeviceBufferMemoryRequirements() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ |
| H A 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
|
| /aosp_15_r20/external/cronet/net/disk_cache/ |
| H A D | disk_cache_fuzzer.proto | 92 required bool is_sparse = 5; field
108 required bool is_sparse = 6; field
|
| H A D | disk_cache_fuzzer.cc | 487 bool is_sparse = ce.is_sparse(); in RunCommands() local
571 bool is_sparse = ooce.is_sparse(); in RunCommands() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v2_sequence_feature_test.py | 29 def test_sequence_feature(self, is_sparse): argument
|
| H A D | tpu_embedding_v2_valid_input_test.py | 247 def test_missing_feature(self, is_sparse): argument
|
| /aosp_15_r20/external/executorch/exir/operator/ |
| H A D | manip.py | 25 is_sparse: bool = False variable in ScratchTensorMetadata
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_csr.py | 2095 def test1(*, is_sparse): argument
2104 def test2(*, is_sparse): argument
2113 def test3(*, is_sparse): argument
2135 def test1(*, is_sparse): argument
2144 def test2(*, is_sparse): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | matmul_op.cc | 35 explicit MatMulOp(OpKernelConstruction* ctx, bool is_sparse = false) in MatMulOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | proximal_gradient_descent_test.py | 135 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
|
| H A D | proximal_adagrad_test.py | 172 def applyOptimizer(self, opt, steps=5, is_sparse=False): argument
|
| H A D | ftrl_test.py | 371 def applyOptimizer(self, opt, dtype, steps=5, is_sparse=False): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | DeprecatedTypeProperties.h | 33 bool is_sparse() const { in is_sparse() function
|
| H A D | TensorBase.h | 487 bool is_sparse() const { in is_sparse() function
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _linalg_utils.py | 10 def is_sparse(A): function
|
| H A D | storage.py | 39 is_sparse: _bool = False variable in _StorageBase
653 is_sparse: _bool = False variable in TypedStorage
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_types.cpp | 21 static const char* parse_privateuseone_backend(bool is_sparse = false) { in parse_privateuseone_backend()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | handle_sparsetensors.cc | 140 const auto is_sparse = main_func.getArgAttrOfType<mlir::BoolAttr>( in CreateComponentTensorsFromSparseTensors() local
|
| H A D | collectives.cc | 218 bool is_sparse = IsSparseValue(input); in EmitRelayout() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/ |
| H A D | tf_utils.py | 393 def is_sparse(tensor): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | sparse_tensor.py | 468 def is_sparse(x): function
|
| /aosp_15_r20/external/pytorch/torch/masked/maskedtensor/ |
| H A D | core.py | 358 def is_sparse(self): member in MaskedTensor
|
| /aosp_15_r20/external/mesa3d/src/nouveau/compiler/ |
| H A D | nak_nir_lower_tex.c | 334 bool is_sparse = false; in shrink_image_load() local
|
| H A D | nak_private.h | 158 bool is_sparse:1; member
|