Searched refs:ValueKnown (Results 1 – 20 of 20) sorted by relevance
84 if (c->ValueKnown(n_dim)) { in PadShapeFn()94 if (c->ValueKnown(n_dim)) { in PadShapeFn()122 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()131 } else if (c->ValueKnown(perm_elems)) { in TransposeShapeFn()198 if (!c->ValueKnown(known_out_elems)) { in SetOutputShapeForReshape()202 if (!c->ValueKnown(dim)) { in SetOutputShapeForReshape()216 if (!c->ValueKnown(known_in_elems)) { in SetOutputShapeForReshape()220 if (!c->ValueKnown(dim)) { in SetOutputShapeForReshape()574 if (!c->ValueKnown(split_dimension)) { in __anondb9326b20a02()621 } else if (size_splits == nullptr && c->ValueKnown(split_dimension)) { in __anondb9326b20b02()[all …]
123 } else if (c->ValueKnown(input1_rank_dim)) { in __anon9b840f3f0202()165 if (c->ValueKnown(input0_rank_dim)) { in __anon9b840f3f0302()174 } else if (c->ValueKnown(input1_rank_dim)) { in __anon9b840f3f0302()
160 if (c->ValueKnown(d) && c->ValueKnown(class_dim)) { in CombinedNMSShapeFn()171 if (c->ValueKnown(output_dim) && c->Value(output_dim) <= 0) { in CombinedNMSShapeFn()183 if (c->ValueKnown(size_per_class) && c->Value(size_per_class) <= 0) { in CombinedNMSShapeFn()576 if (c->ValueKnown(c->Dim(images, 3))) { in __anonc970e8b60f02()
67 if (c->ValueKnown(dim)) { in RaggedRangeShapeFn()
45 if (c->ValueKnown(d)) { in FractionalPoolShapeFn()924 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anon4650176f1d02()925 !c->ValueKnown(filter_rows_dim) || !c->ValueKnown(filter_cols_dim)) { in __anon4650176f1d02()1217 if (c->ValueKnown(last_dim) && c->ValueKnown(k_dim) && in TopKShapeFn()1273 if (c->ValueKnown(last_dim) && c->ValueKnown(n_dim) && in __anon4650176f2602()
82 if (!c->ValueKnown(input_length)) { in SpectrogramShapeFn()
697 if (c->ValueKnown(dim_handle)) { in __anonfa0360991402()
295 if (ValueKnown(dim)) { in ShapeHandleToProto()306 if (!ValueKnown(Dim(s, i))) return false; in FullyDefined()344 return ValueKnown(d) ? strings::StrCat(Value(d)) : "?"; in DebugString()441 } else if (!ValueKnown(d_old) && !ValueKnown(d_new)) { in Relax()449 } else if (!ValueKnown(d_new)) { in Relax()468 } else if (!ValueKnown(d1)) { in Merge()472 } else if (!ValueKnown(d0)) { in Merge()834 if (!ValueKnown(shape_dim)) { in InternalMakeShapeFromTensor()1018 } else if (!ValueKnown(dividend) || in Divide()1019 (divisor.dim.IsSet() && !ValueKnown(divisor.dim))) { in Divide()
1338 if (!c->ValueKnown(dim_x) || !c->ValueKnown(dim_y)) { in BroadcastBinaryOpOutputShapeFnHelper()1429 if (c->ValueKnown(ndims)) { in SliceShape()1504 if (c->ValueKnown(num_index_elements_dim)) { in ValidateSparseTensor()1506 if (c->ValueKnown(num_values_elements_dim)) { in ValidateSparseTensor()1519 if (c->ValueKnown(index_rank_dim)) { in ValidateSparseTensor()1521 if (c->ValueKnown(shape_rank_dim)) { in ValidateSparseTensor()1562 if (c->ValueKnown(index_size)) { in ScatterNdUpdateShape()
177 } else if (c.ValueKnown(out_dim)) { in InferShapes()
367 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown() function
277 EXPECT_TRUE(c.ValueKnown(d)); in TEST_F()281 EXPECT_FALSE(c.ValueKnown(d)); in TEST_F()287 EXPECT_TRUE(c.ValueKnown(d)); in TEST_F()
155 EXPECT_FALSE(c.ValueKnown(c.Dim(output, 1))); in TEST()
36 if (!shape_inference::InferenceContext::ValueKnown(num_input_elements)) { in __anon7c7f74300102()
45 if (c->ValueKnown(tree_depth)) { in __anon3ff10bbb0102()
40 if (c->ValueKnown(d)) { in ProtoFromShapeHandle()
478 CHECK(context->ValueKnown(num_elements_dim)); in RegisterConstantNode()623 if (context->ValueKnown(dim_handle)) { in IsNodeFlattenReshape()925 if (context != nullptr && context->ValueKnown(context->NumElements( in AppendNodeOutputParams()
105 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()1397 if (ic->ValueKnown(size)) { in MaybeUpdateNodeContextOutput()1463 ic->ValueKnown(ic->Dim(shape_handle, 0))) { in MaybeUpdateNodeContextOutput()
505 CHECK(context->ValueKnown(dh)); in BuildTensorShapeMapFromGraph()
1187 if (ic->ValueKnown(dim)) { in TF_GraphGetTensorShape()