Searched refs:output_props (Results 1 – 3 of 3) sorted by relevance
201 output_props = properties.GetOutputProperties(node_name); in PYBIND11_MODULE() local204 prop.reserve(output_props.size()); in PYBIND11_MODULE()205 for (const auto& output_prop : output_props) { in PYBIND11_MODULE()
759 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeReductionIndices() local761 if (output_props.size() != 1) { in MaterializeReductionIndices()764 const OpInfo::TensorProperties& output_prop = output_props[0]; in MaterializeReductionIndices()836 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeConstantValuedNode() local838 if (output_props.size() != 1) return Status::OK(); in MaterializeConstantValuedNode()839 const auto& output_shape = output_props[0].shape(); in MaterializeConstantValuedNode()844 const auto output_dtype = output_props[0].dtype(); in MaterializeConstantValuedNode()1004 const std::vector<OpInfo::TensorProperties>& output_props = in IsFoldableUncached() local1015 for (const auto& output_prop : output_props) { in IsFoldableUncached()1178 auto output_props = properties.GetOutputProperties(node.name()); in GetDataTypeFromNodeOrProps() local[all …]
74 const std::vector<OpInfo::TensorProperties>& output_props, int output_index, in GetOutputDataType() argument76 int output_props_size = output_props.size(); in GetOutputDataType()79 " size of output_props ", output_props.size()); in GetOutputDataType()81 *dtype = output_props[output_index].dtype(); in GetOutputDataType()