Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dshape_optimizer.cc109 const auto& input_props = properties.GetInputProperties(node.name()); in Optimize() local
H A Dconstant_folding.cc740 const std::vector<OpInfo::TensorProperties>& input_props = in MaterializeReductionIndices() local
1006 const std::vector<OpInfo::TensorProperties>& input_props = in IsFoldableUncached() local
2564 const std::vector<OpInfo::TensorProperties>& input_props = in SimplifySelect() local
2803 const auto& input_props = properties.GetInputProperties(node.name())[0]; in IsReductionCandidateForSimplification() local
3564 const auto& input_props = properties.GetInputProperties(conv_node->name()); in MulConvPushDown() local
H A Darithmetic_optimizer.cc469 const OpInfo::TensorProperties* input_props; in HasAllInputsBroadcastableToShape() local
2073 const OpInfo::TensorProperties* input_props; in InputMatchesTargetShape() local
2946 const OpInfo::TensorProperties* input_props; in TrySimplify() local
H A Dremapper.cc474 const std::vector<OpInfo::TensorProperties>& input_props = in IsGpuCompatible() local
1568 auto input_props = in FindMklLayerNorm() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties_test.cc1073 const auto input_props = properties.GetInputProperties("init_restore"); in TEST_F() local