Searched refs:output_props (Results 1 – 6 of 6) 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()
10 for (const auto& output_prop : output_props) {
763 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeReductionIndices() local765 if (output_props.size() != 1) { in MaterializeReductionIndices()768 const OpInfo::TensorProperties& output_prop = output_props[0]; in MaterializeReductionIndices()840 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeConstantValuedNode() local842 if (output_props.size() != 1) return OkStatus(); in MaterializeConstantValuedNode()843 const auto& output_shape = output_props[0].shape(); in MaterializeConstantValuedNode()848 const auto output_dtype = output_props[0].dtype(); in MaterializeConstantValuedNode()1008 const std::vector<OpInfo::TensorProperties>& output_props = in IsFoldableUncached() local1019 for (const auto& output_prop : output_props) { in IsFoldableUncached()1189 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()
2955 const OpInfo::TensorProperties* output_props; in TrySimplify() local2956 TF_RETURN_IF_ERROR(GetTensorProperties(node->name(), &output_props)); in TrySimplify()2957 const TensorShapeProto& output_shape = output_props->shape(); in TrySimplify()
1572 auto output_props = in FindMklLayerNorm() local1575 output_props[0].shape())) { in FindMklLayerNorm()