Searched defs:with_dtype (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| D | _masked.py | 41 with_dtype=False, argument
335 op_info, device, dtype, requires_grad, with_dtype=False, **kwargs argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | dynamic_ragged_shape.py | 981 def with_dtype(self, dtype): member in DynamicRaggedShape
1434 def with_dtype(self, dtype: dtypes.DType) -> "DynamicRaggedShape.Spec": member in DynamicRaggedShape.Spec
1928 def with_dtype(self, dtype): member in _LayerBroadcaster
2074 def with_dtype(self, dtype): member in _GatherLayerBroadcaster
2136 def with_dtype(self, dtype): member in _Broadcaster
|
| D | row_partition.py | 963 def with_dtype(self, dtype): member in RowPartition
1370 def with_dtype(self, dtype): member in RowPartitionSpec
|