Home
last modified time | relevance | path

Searched refs:input_prop (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc313 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources() local
314 if (HasAnyUnknownDimensions(input_prop.shape())) { in VerboseLogUnknownDimensionSources()
327 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources() local
328 inputs += PartialTensorShape::DebugString(input_prop.shape()); in VerboseLogUnknownDimensionSources()
Dgraph_properties_test.cc869 const OpInfo::TensorProperties& input_prop = input_props[1]; in TEST_F() local
870 EXPECT_EQ(DT_FLOAT, input_prop.dtype()); in TEST_F()
871 EXPECT_EQ("float: [128,256]", PropToString(input_prop)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc686 const OpInfo::TensorProperties& input_prop = input_props[0]; in MaterializeReductionIndices() local
687 if (input_prop.shape().unknown_rank()) { in MaterializeReductionIndices()
691 const int input_rank = input_prop.shape().dim_size(); in MaterializeReductionIndices()