Home
last modified time | relevance | path

Searched refs:ValueKnown (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dtpu_cross_replica_ops.cc55 if (c->ValueKnown(num_replicas_per_group) && in __anond991e9790102()
76 if (!c->ValueKnown(c->Dim(input, concat_dimension)) || in __anond991e9790102()
77 !c->ValueKnown(c->Dim(input, split_dimension))) { in __anond991e9790102()
91 if (c->ValueKnown(dims[i]) && in __anond991e9790102()
Dset_ops.cc123 } else if (c->ValueKnown(input1_rank_dim)) { in __anon1e42d5040202()
165 if (c->ValueKnown(input0_rank_dim)) { in __anon1e42d5040302()
174 } else if (c->ValueKnown(input1_rank_dim)) { in __anon1e42d5040302()
Darray_ops.cc92 if (c->ValueKnown(n_dim)) { in PadShapeFn()
102 if (c->ValueKnown(n_dim)) { in PadShapeFn()
130 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()
139 } else if (c->ValueKnown(perm_elems)) { in TransposeShapeFn()
205 if (!c->ValueKnown(known_out_elems)) { in SetOutputShapeForReshape()
209 if (!c->ValueKnown(dim)) { in SetOutputShapeForReshape()
223 if (!c->ValueKnown(known_in_elems)) { in SetOutputShapeForReshape()
227 if (!c->ValueKnown(dim)) { in SetOutputShapeForReshape()
581 if (!c->ValueKnown(split_dimension)) { in __anon091a71b70a02()
628 } else if (size_splits == nullptr && c->ValueKnown(split_dimension)) { in __anon091a71b70b02()
[all …]
Drisc_ops.cc31 if (!c->ValueKnown(c->Dim(c->input(0), i)) || in RiscBinaryNonBroadcastOpShapeFn()
32 !c->ValueKnown(c->Dim(c->input(1), i))) { in RiscBinaryNonBroadcastOpShapeFn()
Dragged_math_ops.cc68 if (c->ValueKnown(dim)) { in RaggedRangeShapeFn()
Dcount_ops.cc30 if (c->ValueKnown(num_weights) && c->Value(num_weights) == 0) { in DenseCountSparseOutputShapeFn()
Dimage_ops.cc220 if (c->ValueKnown(d) && c->ValueKnown(class_dim)) { in CombinedNMSShapeFn()
231 if (c->ValueKnown(output_dim) && c->Value(output_dim) <= 0) { in CombinedNMSShapeFn()
243 if (c->ValueKnown(size_per_class) && c->Value(size_per_class) <= 0) { in CombinedNMSShapeFn()
660 if (c->ValueKnown(c->Dim(images, 3))) { in __anonf6f833bb0f02()
Dlinalg_ops.cc61 if (c->ValueKnown(num_bands) && c->Value(num_bands) <= 0) { in BandedTriangularSolveShapeFn()
65 if (c->ValueKnown(num_bands) && c->ValueKnown(m) && in BandedTriangularSolveShapeFn()
Daudio_ops.cc82 if (!c->ValueKnown(input_length)) { in SpectrogramShapeFn()
Dnn_ops.cc51 if (c->ValueKnown(d)) { in FractionalPoolShapeFn()
1052 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anon1bbab8941702()
1053 !c->ValueKnown(filter_rows_dim) || !c->ValueKnown(filter_cols_dim)) { in __anon1bbab8941702()
1345 if (c->ValueKnown(last_dim) && c->ValueKnown(k_dim) && in TopKShapeFn()
1496 if (c->ValueKnown(last_dim) && c->ValueKnown(n_dim) && in __anon1bbab8942102()
Dsparse_csr_matrix_ops.cc133 if (!c->ValueKnown(indices_col) || c->Value(indices_col) != rank) { in __anon6adce4610302()
Ddataset_ops.cc1143 if (c->ValueKnown(dim_handle)) { in __anona68980de2102()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc245 if (ValueKnown(dim)) { in ShapeHandleToProto()
256 if (!ValueKnown(Dim(s, i))) return false; in FullyDefined()
294 return ValueKnown(d) ? strings::StrCat(Value(d)) : "?"; in DebugString()
390 } else if (!ValueKnown(d_old) && !ValueKnown(d_new)) { in Relax()
398 } else if (!ValueKnown(d_new)) { in Relax()
417 } else if (!ValueKnown(d1)) { in Merge()
421 } else if (!ValueKnown(d0)) { in Merge()
785 if (!ValueKnown(shape_dim)) { in InternalMakeShapeFromTensor()
1024 } else if (!ValueKnown(dividend) || in Divide()
1025 (divisor.dim.IsSet() && !ValueKnown(divisor.dim))) { in Divide()
Dcommon_shape_fns.cc672 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv2DShapeImpl()
684 if (c->ValueKnown(output_depth_dim)) { in Conv2DShapeImpl()
835 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv3DShape()
847 if (c->ValueKnown(output_depth_dim)) { in Conv3DShape()
1251 if (c->ValueKnown(channel_dim) && c->Value(channel_dim) % 4 != 0) { in FusedBatchNormExShape()
2109 if (!c->ValueKnown(dim_x) || !c->ValueKnown(dim_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
2137 } else if (!c->ValueKnown(dim_x) && !c->ValueKnown(dim_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
2253 if (c->ValueKnown(ndims)) { in SliceShape()
2328 if (c->ValueKnown(num_index_elements_dim)) { in ValidateSparseTensor()
2330 if (c->ValueKnown(num_values_elements_dim)) { in ValidateSparseTensor()
[all …]
Dshape_inference_testutil.cc175 } else if (c.ValueKnown(out_dim)) { in InferShapes()
Dcommon_shape_fns_test.cc157 EXPECT_FALSE(c.ValueKnown(c.Dim(output, 1))); in TEST()
1544 EXPECT_FALSE(c.ValueKnown(c.Dim(output, 0))); in TEST()
1545 EXPECT_FALSE(c.ValueKnown(c.Dim(output, 1))); in TEST()
Dshape_inference.h437 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown() function
Dshape_inference_test.cc302 EXPECT_TRUE(c.ValueKnown(d)); in TEST_F()
306 EXPECT_FALSE(c.ValueKnown(d)); in TEST_F()
312 EXPECT_TRUE(c.ValueKnown(d)); in TEST_F()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_meta_ops.cc81 if (!c->ValueKnown(dim) || in __anon82f11ae70102()
138 if (!c->ValueKnown(dim) || in __anon82f11ae70202()
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc93 if (shape_c->ValueKnown(shape_c->Dim(shape_c->input(0), i))) { in TryToInferTensorOutputFromShapeNode()
108 if (shape_c->ValueKnown(shape_c->Dim(shape_c->input(0), i))) { in TryToInferTensorOutputFromShapeNode()
/external/tensorflow/tensorflow/c/
Dops.cc244 return InferenceContext::ValueKnown( in TF_DimensionHandleValueKnown()
Dc_api.cc765 if (ic->ValueKnown(dim)) { in TF_GraphGetTensorShape()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc116 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()
380 if (ic->ValueKnown(dim) && ic->Value(dim) == kUnknownDimFromConst) { in ReplaceUnknownDimFromConstWithUnknownDim()
683 if (ic->ValueKnown(d) && ic->Value(d) == kUnknownDimFromConst) { in StringifyShapeHandle()
1439 if (ic->ValueKnown(dim) && ic->Value(dim) == kUnknownDimFromConst) { in AllOutputValuesKnown()
1731 if (ic->ValueKnown(size)) { in MaybeUpdateNodeContextOutput()
1799 ic->ValueKnown(ic->Dim(shape_handle, 0))) { in MaybeUpdateNodeContextOutput()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc542 if (c->ValueKnown(orig_size_handle)) { in __anonb720852f0302()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1273 if (!context->ValueKnown(dim_handle)) in ConvertElementTypeAndShape()