Searched refs:output_prop (Results 1 – 3 of 3) sorted by relevance
205 for (const auto& output_prop : output_props) { in PYBIND11_MODULE() local206 prop.push_back(output_prop.SerializeAsString()); in PYBIND11_MODULE()
764 const OpInfo::TensorProperties& output_prop = output_props[0]; in MaterializeReductionIndices() local766 output_prop.shape().unknown_rank() ? -1 : output_prop.shape().dim_size(); in MaterializeReductionIndices()1015 for (const auto& output_prop : output_props) { in IsFoldableUncached() local1016 const PartialTensorShape output_shape(output_prop.shape()); in IsFoldableUncached()1019 output_shape.num_elements() * DataTypeSize(output_prop.dtype()); in IsFoldableUncached()
331 for (const auto& output_prop : output_properties) { in VerboseLogUnknownDimensionSources() local332 if (HasAnyUnknownDimensions(output_prop.shape())) { in VerboseLogUnknownDimensionSources()340 for (const auto& output_prop : output_properties) { in VerboseLogUnknownDimensionSources() local341 outputs += PartialTensorShape::DebugString(output_prop.shape()); in VerboseLogUnknownDimensionSources()