Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/
H A 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
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A Dmetadata.py95 tensor_properties: TensorProperties = field(default_factory=TensorProperties) variable in ShardedTensorMetadata
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dbuiltin_functions.cpp50 auto tensor_properties = variable
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc2776 auto tensor_properties = GetOutputProperties(node->name()); in AnnotateOutputShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc931 const auto& tensor_properties = output_params.at(output_port); in ConvertToTensorOrWeights() local