Home
last modified time | relevance | path

Searched defs:from_tensor (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_conversion_ops.py26 def from_tensor(tensor, function
Dragged_tensor_shape.py177 def from_tensor(cls, rt_input, dim_size_dtype=None): member in RaggedTensorDynamicShape
Dragged_tensor.py1575 def from_tensor(cls, member in RaggedTensor
Ddynamic_ragged_shape.py430 def from_tensor(cls, t, dtype=None): member in DynamicRaggedShape
/external/tensorflow/tensorflow/python/keras/engine/
Dkeras_tensor.py144 def from_tensor(cls, tensor): member in KerasTensor
523 def from_tensor(cls, tensor): member in UserRegisteredTypeKerasTensor
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc100 const AllocatorAttributes& from_alloc_attr, const Tensor* from_tensor, in PostToPeer()
Dcollective_test_util.cc75 const Tensor* from_tensor, in PostToPeer()
Dgraph_execution_state.cc314 NodeBuilder::NodeOut from_tensor) in TensorConnectionPruneRewrite()
/external/executorch/exir/
Dtensor.py163 def from_tensor(cls, tensor: torch.Tensor, const: bool = False) -> TensorSpec: member in TensorSpec
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py281 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): argument
/external/tensorflow/tensorflow/python/framework/
Dtensor_spec.py166 def from_tensor(cls, tensor, name=None): member in TensorSpec
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto831 string from_tensor = 1; field
/external/pytorch/torch/_subclasses/
Dfake_tensor.py743 def from_tensor(t: Tensor, fake_mode: FakeTensorMode) -> FakeTensor: member in FakeTensor
2221 def from_tensor( member in FakeTensorMode
/external/pytorch/torch/profiler/
D_memory_profiler.py142 def from_tensor(cls, t: Optional[_TensorMetadata]) -> Optional["TensorKey"]: member in TensorKey
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc335 SafeTensorId from_tensor(func_node, from_idx); in AddTensorMapping() local