Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc123 } else if (c->ValueKnown(input1_rank_dim)) { in __anonda6063e60202()
165 if (c->ValueKnown(input0_rank_dim)) { in __anonda6063e60302()
174 } else if (c->ValueKnown(input1_rank_dim)) { in __anonda6063e60302()
Darray_ops.cc91 if (c->ValueKnown(n_dim)) { in PadShapeFn()
101 if (c->ValueKnown(n_dim)) { in PadShapeFn()
129 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()
138 } 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 __anon42d741190a02()
627 } else if (size_splits == nullptr && c->ValueKnown(split_dimension)) { in __anon42d741190b02()
[all …]
Dragged_math_ops.cc68 if (c->ValueKnown(dim)) { in RaggedRangeShapeFn()
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()
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 __anon30b5031d0f02()
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.cc48 if (c->ValueKnown(d)) { in FractionalPoolShapeFn()
1018 if (!c->ValueKnown(in_rows_dim) || !c->ValueKnown(in_cols_dim) || in __anonad10f3361702()
1019 !c->ValueKnown(filter_rows_dim) || !c->ValueKnown(filter_cols_dim)) { in __anonad10f3361702()
1311 if (c->ValueKnown(last_dim) && c->ValueKnown(k_dim) && in TopKShapeFn()
1367 if (c->ValueKnown(last_dim) && c->ValueKnown(n_dim) && in __anonad10f3362002()
Dsparse_csr_matrix_ops.cc133 if (!c->ValueKnown(indices_col) || c->Value(indices_col) != rank) { in __anon59a9d6c30302()
Ddataset_ops.cc952 if (c->ValueKnown(dim_handle)) { in __anon42b7b0c02002()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc238 if (ValueKnown(dim)) { in ShapeHandleToProto()
249 if (!ValueKnown(Dim(s, i))) return false; in FullyDefined()
287 return ValueKnown(d) ? strings::StrCat(Value(d)) : "?"; in DebugString()
383 } else if (!ValueKnown(d_old) && !ValueKnown(d_new)) { in Relax()
391 } else if (!ValueKnown(d_new)) { in Relax()
410 } else if (!ValueKnown(d1)) { in Merge()
414 } else if (!ValueKnown(d0)) { in Merge()
778 if (!ValueKnown(shape_dim)) { in InternalMakeShapeFromTensor()
991 } else if (!ValueKnown(dividend) || in Divide()
992 (divisor.dim.IsSet() && !ValueKnown(divisor.dim))) { in Divide()
Dcommon_shape_fns.cc664 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv2DShapeImpl()
674 if (c->ValueKnown(output_depth_dim)) { in Conv2DShapeImpl()
803 if (c->ValueKnown(input_depth_dim) && c->ValueKnown(filter_input_depth_dim)) { in Conv3DShape()
813 if (c->ValueKnown(output_depth_dim)) { in Conv3DShape()
1196 if (c->ValueKnown(channel_dim) && c->Value(channel_dim) % 4 != 0) { in FusedBatchNormExShape()
2018 if (!c->ValueKnown(dim_x) || !c->ValueKnown(dim_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
2046 } else if (!c->ValueKnown(dim_x) && !c->ValueKnown(dim_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
2162 if (c->ValueKnown(ndims)) { in SliceShape()
2237 if (c->ValueKnown(num_index_elements_dim)) { in ValidateSparseTensor()
2239 if (c->ValueKnown(num_values_elements_dim)) { in ValidateSparseTensor()
[all …]
Dshape_inference_testutil.cc175 } else if (c.ValueKnown(out_dim)) { in InferShapes()
Dshape_inference.h353 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown() function
Dshape_inference_test.cc294 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()
Dcommon_shape_fns_test.cc155 EXPECT_FALSE(c.ValueKnown(c.Dim(output, 1))); in TEST()
/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.cc112 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()
376 if (ic->ValueKnown(dim) && ic->Value(dim) == kUnknownDimFromConst) { in ReplaceUnknownDimFromConstWithUnknownDim()
654 if (ic->ValueKnown(d) && ic->Value(d) == kUnknownDimFromConst) { in StringifyShapeHandle()
1400 if (ic->ValueKnown(dim) && ic->Value(dim) == kUnknownDimFromConst) { in AllOutputValuesKnown()
1692 if (ic->ValueKnown(size)) { in MaybeUpdateNodeContextOutput()
1760 ic->ValueKnown(ic->Dim(shape_handle, 0))) { in MaybeUpdateNodeContextOutput()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc478 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()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc466 if (c->ValueKnown(orig_size_handle)) { in __anonbbc1c2d10302()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc506 CHECK(context->ValueKnown(dh)); in BuildTensorShapeMapFromGraph()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1126 if (!context->ValueKnown(dim_handle)) in ConvertElementTypeAndShape()