Searched refs:ValueKnown (Results 1 – 20 of 20) sorted by relevance
123 } else if (c->ValueKnown(input1_rank_dim)) { in __anon0af7a88a0202()165 if (c->ValueKnown(input0_rank_dim)) { in __anon0af7a88a0302()174 } else if (c->ValueKnown(input1_rank_dim)) { in __anon0af7a88a0302()
90 if (c->ValueKnown(n_dim)) { in PadShapeFn()100 if (c->ValueKnown(n_dim)) { in PadShapeFn()128 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()137 } else if (c->ValueKnown(perm_elems)) { in TransposeShapeFn()204 if (!c->ValueKnown(known_out_elems)) { in SetOutputShapeForReshape()208 if (!c->ValueKnown(dim)) { in SetOutputShapeForReshape()222 if (!c->ValueKnown(known_in_elems)) { in SetOutputShapeForReshape()226 if (!c->ValueKnown(dim)) { in SetOutputShapeForReshape()580 if (!c->ValueKnown(split_dimension)) { in __anonf6523ebd0a02()627 } else if (size_splits == nullptr && c->ValueKnown(split_dimension)) { in __anonf6523ebd0b02()[all …]
68 if (c->ValueKnown(dim)) { in RaggedRangeShapeFn()
186 if (c->ValueKnown(d) && c->ValueKnown(class_dim)) { in CombinedNMSShapeFn()197 if (c->ValueKnown(output_dim) && c->Value(output_dim) <= 0) { in CombinedNMSShapeFn()209 if (c->ValueKnown(size_per_class) && c->Value(size_per_class) <= 0) { in CombinedNMSShapeFn()609 if (c->ValueKnown(c->Dim(images, 3))) { in __anone43000c10f02()
82 if (!c->ValueKnown(input_length)) { in SpectrogramShapeFn()
46 if (c->ValueKnown(d)) { in FractionalPoolShapeFn()989 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anon87c0335a1d02()990 !c->ValueKnown(filter_rows_dim) || !c->ValueKnown(filter_cols_dim)) { in __anon87c0335a1d02()1282 if (c->ValueKnown(last_dim) && c->ValueKnown(k_dim) && in TopKShapeFn()1338 if (c->ValueKnown(last_dim) && c->ValueKnown(n_dim) && in __anon87c0335a2602()
818 if (c->ValueKnown(dim_handle)) { in __anonc0e8a7641a02()
133 if (!c->ValueKnown(indices_col) || c->Value(indices_col) != rank) { in __anond96a6a670302()
236 if (ValueKnown(dim)) { in ShapeHandleToProto()247 if (!ValueKnown(Dim(s, i))) return false; in FullyDefined()285 return ValueKnown(d) ? strings::StrCat(Value(d)) : "?"; in DebugString()381 } else if (!ValueKnown(d_old) && !ValueKnown(d_new)) { in Relax()389 } else if (!ValueKnown(d_new)) { in Relax()408 } else if (!ValueKnown(d1)) { in Merge()412 } else if (!ValueKnown(d0)) { in Merge()774 if (!ValueKnown(shape_dim)) { in InternalMakeShapeFromTensor()958 } else if (!ValueKnown(dividend) || in Divide()959 (divisor.dim.IsSet() && !ValueKnown(divisor.dim))) { in Divide()
752 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv2DShapeImpl()762 if (c->ValueKnown(output_depth_dim)) { in Conv2DShapeImpl()891 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv3DShape()901 if (c->ValueKnown(output_depth_dim)) { in Conv3DShape()1138 if (c->ValueKnown(channel_dim) && c->Value(channel_dim) % 4 != 0) { in FusedBatchNormExShape()1915 if (!c->ValueKnown(dim_x) || !c->ValueKnown(dim_y)) { in BroadcastBinaryOpOutputShapeFnHelper()2056 if (c->ValueKnown(ndims)) { in SliceShape()2131 if (c->ValueKnown(num_index_elements_dim)) { in ValidateSparseTensor()2133 if (c->ValueKnown(num_values_elements_dim)) { in ValidateSparseTensor()2146 if (c->ValueKnown(index_rank_dim)) { in ValidateSparseTensor()[all …]
175 } else if (c.ValueKnown(out_dim)) { in InferShapes()
332 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown() function
294 EXPECT_TRUE(c.ValueKnown(d)); in TEST_F()298 EXPECT_FALSE(c.ValueKnown(d)); in TEST_F()304 EXPECT_TRUE(c.ValueKnown(d)); in TEST_F()
155 EXPECT_FALSE(c.ValueKnown(c.Dim(output, 1))); in TEST()
238 return InferenceContext::ValueKnown( in TF_DimensionHandleValueKnown()
763 if (ic->ValueKnown(dim)) { in TF_GraphGetTensorShape()
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()
106 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()1507 if (ic->ValueKnown(size)) { in MaybeUpdateNodeContextOutput()1573 ic->ValueKnown(ic->Dim(shape_handle, 0))) { in MaybeUpdateNodeContextOutput()
505 CHECK(context->ValueKnown(dh)); in BuildTensorShapeMapFromGraph()
914 if (!context->ValueKnown(dim_handle)) in ConvertElementTypeAndShape()