Home
last modified time | relevance | path

Searched refs:reshape_props (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1984 const OpInfo::TensorProperties* reshape_props; in InputMatchesTargetShape() local
1987 if (!GetTensorProperties(reshape.name(), &reshape_props).ok() || in InputMatchesTargetShape()
1993 reshape_props->shape()); in InputMatchesTargetShape()
Dconstant_folding.cc779 const std::vector<OpInfo::TensorProperties>& reshape_props = in MaterializeReductionIndices() local
781 if (reshape_props.size() != 1) { in MaterializeReductionIndices()
784 const OpInfo::TensorProperties& reshape_prop = reshape_props[0]; in MaterializeReductionIndices()