Home
last modified time | relevance | path

Searched defs:tensor_properties (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/distributed/_shard/sharded_tensor/
Dmetadata.py95 tensor_properties: TensorProperties = field(default_factory=TensorProperties) variable in ShardedTensorMetadata
/external/pytorch/torch/csrc/jit/frontend/
Dbuiltin_functions.cpp50 auto tensor_properties = variable
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler.cc310 const std::vector<tensorflow::OpInfo::TensorProperties>& tensor_properties = in TF_GetInputPropertiesList() local
329 const std::vector<tensorflow::OpInfo::TensorProperties>& tensor_properties = in TF_GetOutputPropertiesList() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2776 auto tensor_properties = GetOutputProperties(node->name()); in AnnotateOutputShapes() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc931 const auto& tensor_properties = output_params.at(output_port); in ConvertToTensorOrWeights() local