Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/grappler/
Ditem_wrapper.cc201 output_props = properties.GetOutputProperties(node_name); in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc763 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeReductionIndices() local
840 const std::vector<OpInfo::TensorProperties>& output_props = in MaterializeConstantValuedNode() local
1008 const std::vector<OpInfo::TensorProperties>& output_props = in IsFoldableUncached() local
1189 auto output_props = properties.GetOutputProperties(node.name()); in GetDataTypeFromNodeOrProps() local
2804 const auto& output_props = properties.GetOutputProperties(node.name())[0]; in IsReductionCandidateForSimplification() local
Dscoped_allocator_optimizer.cc74 const std::vector<OpInfo::TensorProperties>& output_props, int output_index, in GetOutputDataType()
Dremapper.cc1572 auto output_props = in FindMklLayerNorm() local
Darithmetic_optimizer.cc2955 const OpInfo::TensorProperties* output_props; in TrySimplify() local