Home
last modified time | relevance | path

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

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