Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc50 if (!c->RankKnown(matrix_shape)) { in ValidateSquareMatrixShape()
72 if (!c->RankKnown(dense_shape) || c->Rank(dense_shape) < 2 || in __anon6adce4610102()
98 if (!c->RankKnown(sparse_matrix)) { in __anon6adce4610202()
118 if (!c->RankKnown(dense_shape) || c->Rank(dense_shape) < 2 || in __anon6adce4610302()
127 if (!c->RankKnown(indices) || c->Rank(indices) != 2) { in __anon6adce4610302()
160 if (!c->RankKnown(sparse_matrix)) { in __anon6adce4610402()
186 if (!c->RankKnown(csr_sparse_matrix)) { in __anon6adce4610502()
208 if (!c->RankKnown(sparse_matrix)) { in __anon6adce4610602()
238 if (!c->RankKnown(a_shape)) { in __anon6adce4610702()
308 if (!c->RankKnown(a_shape)) { in __anon6adce4610802()
[all …]
Dset_ops.cc56 if (c->RankKnown(input0_shape)) { in __anon1e42d5040102()
61 if (c->RankKnown(input1_shape)) { in __anon1e42d5040102()
78 if (c->RankKnown(input1_shape)) { in __anon1e42d5040102()
118 if (c->RankKnown(input0_shape)) { in __anon1e42d5040202()
Darray_ops.cc94 } else if (c->RankKnown(input)) { in PadShapeFn()
130 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()
137 if (c->RankKnown(input)) { in TransposeShapeFn()
144 if (!c->RankKnown(input) && rank < 2) { in TransposeShapeFn()
193 if (!c->RankKnown(out)) { in SetOutputShapeForReshape()
198 if (c->RankKnown(in)) { in SetOutputShapeForReshape()
338 if (!c->RankKnown(cur)) { in __anon091a71b70302()
424 if (c->RankKnown(s)) { in __anon091a71b70502()
459 if (c->RankKnown(indices) && c->Rank(indices) == 0) { in __anon091a71b70602()
461 } else if (c->RankKnown(indices)) { in __anon091a71b70602()
[all …]
Dtpu_cross_replica_ops.cc37 if (!c->RankKnown(input)) { in __anond991e9790102()
51 if (c->RankKnown(group_assignment) && c->Rank(group_assignment) != 2) { in __anond991e9790102()
Dcontrol_flow_ops.cc124 if (!c->RankKnown(out)) { in MergeShape()
130 if (!c->RankKnown(input) || c->Rank(input) != rank) { in MergeShape()
Dragged_conversion_ops.cc177 c->RankKnown(rt_dense_values) in RaggedTensorToSparseShapeFn()
227 if (c->RankKnown(encoded_ragged) && input_ragged_rank >= 0) { in RaggedTensorFromVariantShapeFn()
Dparsing_ops.cc50 if (c->RankKnown(input_shape)) { in AddSparseOutputShapes()
215 if (c->RankKnown(input) && c->Rank(input) == 1) { in __anon5cc8b68c0502()
377 if (c->RankKnown(input) && c->Rank(input) == 1) { in __anon5cc8b68c0802()
Dlogging_ops.cc56 if (!c->RankKnown(c->input(0))) return OkStatus(); in __anon6c2cd2df0102()
Dstring_ops.cc188 if (c->RankKnown(c->input(i)) && c->Rank(c->input(i)) != 0) { in __anond1fe1a4f0402()
278 if (c->RankKnown(len_shape)) { in __anond1fe1a4f0702()
Dlookup_ops.cc131 if (c->RankKnown(key_shape_and_type.shape) && c->RankKnown(keys)) { in ValidateTableResourceHandle()
Dmath_ops.cc692 if (!c->RankKnown(in)) { in __anon328d24420202()
848 if (!c->RankKnown(cond) || !c->RankKnown(data)) { in __anon328d24420402()
1083 if (!c->RankKnown(input_shape)) { in ArgOpShape()
1653 if (c->RankKnown(a_shape)) { in __anon328d24420c02()
Dlist_ops.cc288 if (c->RankKnown(element_shape)) { in TensorListConcatShapeInference()
411 if (c->RankKnown((*handle_data)[0].shape)) { in __anon6c1e81940b02()
Ddata_flow_ops.cc61 if (!c->RankKnown(partitions_shape)) { in __anon89875fc90202()
106 if (!c->RankKnown(indices_shape)) { in DynamicStitchShapeFunction()
691 if (c->RankKnown(input_shape) && in __anon89875fc91702()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc40 if (c->RankKnown(c->input(0))) { in UnchangedRank()
208 if (!c->RankKnown(lhs_shape_handle) || !c->RankKnown(rhs_shape_handle)) { in XlaDotShapeFunction()
387 if (!c->RankKnown(size_indices_shape)) { in __anonb720852f0202()
395 if (!c->RankKnown(size_indices_value)) { in __anonb720852f0202()
490 if (!c->RankKnown(input_shape_handle)) { in __anonb720852f0302()
496 if (c->RankKnown(padding_shape_handle) && in __anonb720852f0302()
611 if (c->RankKnown(c->input(0))) { in __anonb720852f0502()
661 if (c->RankKnown(c->input(0))) { in __anonb720852f0702()
740 if (c->RankKnown(input_shape)) { in __anonb720852f0902()
1044 if (!c->RankKnown(input_handle)) { in __anonb720852f0f02()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc237 if (!RankKnown(handle)) { in ShapeHandleToProto()
254 if (!RankKnown(s)) return false; in FullyDefined()
284 if (RankKnown(s)) { in DebugString()
439 if (!RankKnown(prefix) || !RankKnown(s)) { in MergePrefix()
466 } else if (!RankKnown(s_new) || !s_old.IsSet()) { in Relax()
511 } else if (!RankKnown(s1)) { in Merge()
515 } else if (!RankKnown(s0)) { in Merge()
592 ((RankKnown(s) && end >= rank) || in Subshape()
597 if (!RankKnown(s)) { in Subshape()
647 if (!RankKnown(s1) || !RankKnown(s2)) { in Concatenate()
[all …]
Dcommon_shape_fns.cc187 if (c->RankKnown(input_shape)) { in EinsumShape()
214 c->RankKnown(input_shape) in EinsumShape()
220 if (!c->RankKnown(input_shape)) { in EinsumShape()
235 DimensionHandle new_dim = c->RankKnown(input_shape) in EinsumShape()
269 if (!c->RankKnown(output_bcast_shape)) { in EinsumShape()
400 if (!c->RankKnown(input_shape)) { in BiasAddShape()
1366 if (!c->RankKnown(input_shape) || !c->FullyDefined(diag_index_shape) || in MatrixDiagPartV2Shape()
1423 if (!c->RankKnown(input_shape) || !c->FullyDefined(diag_index_shape) || in MatrixDiagV2Shape()
1534 if (c->RankKnown(input_shape)) { in MatrixSetDiagV2Shape()
1567 if (c->RankKnown(diag_shape) && !c->FullyDefined(input_shape)) { in MatrixSetDiagV2Shape()
[all …]
Dshape_inference_testutil.cc130 if (c.RankKnown(out)) { in InferShapes()
144 if (!c.RankKnown(out)) { in InferShapes()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_meta_ops.cc53 if (!c->RankKnown(in)) { in __anon82f11ae70102()
110 if (!c->RankKnown(in)) { in __anon82f11ae70202()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc97 if (InferenceContext::RankKnown(*result)) { in Merge()
102 if (InferenceContext::RankKnown(h1)) { in Merge()
372 if (!ic->RankKnown(shape)) { in ReplaceUnknownDimFromConstWithUnknownDim()
679 if (ic->RankKnown(s)) { in StringifyShapeHandle()
1187 if (!ctx->RankKnown(shape2) || ctx->Rank(shape2) != rank) { in OutputAsUnion()
1211 if (!InferenceContext::RankKnown(s1) && !InferenceContext::RankKnown(s2)) { in EquivalentShapes()
1242 if (!InferenceContext::RankKnown(inferred_shape)) { in CompatibleShapes()
1721 if (ic->RankKnown(ic->input(0))) { in MaybeUpdateNodeContextOutput()
1764 if (!ic->RankKnown(input)) { in MaybeUpdateNodeContextOutput()
1798 if (ic->RankKnown(shape_handle) && ic->Rank(shape_handle) >= 1 && in MaybeUpdateNodeContextOutput()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc518 if (!target_context->RankKnown(pre_cast_shape)) { in ConstantPartialShape()
564 if (!target_context->RankKnown(sub_result)) { in ConstantPartialShape()
796 if (!c->RankKnown(s0) && !c->RankKnown(s1)) { in SameDefinedShape()
Deval_const_tensor.cc85 if (!shape_c->RankKnown(shape_c->input(0))) return OkStatus(); in TryToInferTensorOutputFromShapeNode()
148 if (!shape_c->RankKnown(shape_c->input(0))) return OkStatus(); in TryToInferTensorOutputFromStridedSliceNode()
236 bool rank_known = c->RankKnown(c->input(0)); in TryToInferTensorOutputFromInputShapes()
/external/tensorflow/tensorflow/lite/kernels/shim/
Dtf_op_shim.cc118 if (!context_->RankKnown(shape)) return Shape(); in GetInputShape()
/external/tensorflow/tensorflow/c/
Dops.cc158 return cc_ctx->RankKnown(*reinterpret_cast<ShapeHandle*>(handle)); in TF_ShapeInferenceContextRankKnown()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc36 if (!context->RankKnown(handle)) return OkStatus(); in ShapeHandleToTensorShape()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc202 if (!context->RankKnown(handle)) return OkStatus(); in ShapeHandleToTensorShape()

12