Searched refs:output_props (Results 1 – 2 of 2) sorted by relevance
157 const std::vector<tensorflow::OpInfo::TensorProperties>& output_props = in TF_GetOpProperties() local160 PyObject* prop = PyList_New(output_props.size()); in TF_GetOpProperties()161 for (int i = 0; i < output_props.size(); ++i) { in TF_GetOpProperties()162 string output_prop_str = output_props[i].SerializeAsString(); in TF_GetOpProperties()
704 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeReductionIndices() local706 if (output_props.size() != 1) { in MaterializeReductionIndices()709 const OpInfo::TensorProperties& output_prop = output_props[0]; in MaterializeReductionIndices()778 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeConstantValuedNode() local780 if (output_props.size() != 1) return Status::OK(); in MaterializeConstantValuedNode()781 const auto& output_shape = output_props[0].shape(); in MaterializeConstantValuedNode()786 const auto output_dtype = output_props[0].dtype(); in MaterializeConstantValuedNode()1019 auto output_props = properties.GetOutputProperties(node.name()); in GetDataTypeFromNodeOrProps() local1020 if (!output_props.empty()) { in GetDataTypeFromNodeOrProps()1021 dtype = output_props[0].dtype(); in GetDataTypeFromNodeOrProps()[all …]