Searched refs:input_prop (Results 1 – 3 of 3) sorted by relevance
741 const OpInfo::TensorProperties& input_prop = input_props[0]; in MaterializeReductionIndices() local742 if (input_prop.shape().unknown_rank()) { in MaterializeReductionIndices()746 const int input_rank = input_prop.shape().dim_size(); in MaterializeReductionIndices()1008 for (const auto& input_prop : input_props) { in IsFoldableUncached() local1009 const PartialTensorShape input_shape(input_prop.shape()); in IsFoldableUncached()1012 input_shape.num_elements() * DataTypeSize(input_prop.dtype()); in IsFoldableUncached()
320 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources() local321 if (HasAnyUnknownDimensions(input_prop.shape())) { in VerboseLogUnknownDimensionSources()334 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources() local335 inputs += PartialTensorShape::DebugString(input_prop.shape()); in VerboseLogUnknownDimensionSources()
907 const OpInfo::TensorProperties& input_prop = input_props[1]; in TEST_F() local908 EXPECT_EQ(DT_FLOAT, input_prop.dtype()); in TEST_F()909 EXPECT_EQ("float: [128,256]", PropToString(input_prop)); in TEST_F()