| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 50 if (!c->RankKnown(matrix_shape)) { in ValidateSquareMatrixShape()
72 if (!c->RankKnown(dense_shape) || c->Rank(dense_shape) < 2 || in __anon59a9d6c30102()
98 if (!c->RankKnown(sparse_matrix)) { in __anon59a9d6c30202()
118 if (!c->RankKnown(dense_shape) || c->Rank(dense_shape) < 2 || in __anon59a9d6c30302()
127 if (!c->RankKnown(indices) || c->Rank(indices) != 2) { in __anon59a9d6c30302()
160 if (!c->RankKnown(sparse_matrix)) { in __anon59a9d6c30402()
186 if (!c->RankKnown(csr_sparse_matrix)) { in __anon59a9d6c30502()
208 if (!c->RankKnown(sparse_matrix)) { in __anon59a9d6c30602()
238 if (!c->RankKnown(a_shape)) { in __anon59a9d6c30702()
308 if (!c->RankKnown(a_shape)) { in __anon59a9d6c30802()
[all …]
|
| D | set_ops.cc | 56 if (c->RankKnown(input0_shape)) { in __anonda6063e60102()
61 if (c->RankKnown(input1_shape)) { in __anonda6063e60102()
78 if (c->RankKnown(input1_shape)) { in __anonda6063e60102()
118 if (c->RankKnown(input0_shape)) { in __anonda6063e60202()
|
| D | array_ops.cc | 93 } else if (c->RankKnown(input)) { in PadShapeFn()
129 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()
136 if (c->RankKnown(input)) { in TransposeShapeFn()
143 if (!c->RankKnown(input) && rank < 2) { in TransposeShapeFn()
192 if (!c->RankKnown(out)) { in SetOutputShapeForReshape()
197 if (c->RankKnown(in)) { in SetOutputShapeForReshape()
337 if (!c->RankKnown(cur)) { in __anon42d741190302()
423 if (c->RankKnown(s)) { in __anon42d741190502()
458 if (c->RankKnown(indices) && c->Rank(indices) == 0) { in __anon42d741190602()
460 } else if (c->RankKnown(indices)) { in __anon42d741190602()
[all …]
|
| D | control_flow_ops.cc | 123 if (!c->RankKnown(out)) { in MergeShape()
129 if (!c->RankKnown(input) || c->Rank(input) != rank) { in MergeShape()
|
| D | logging_ops.cc | 56 if (!c->RankKnown(c->input(0))) return Status::OK(); in __anon085b02c10102()
|
| D | ragged_conversion_ops.cc | 176 c->RankKnown(rt_dense_values) in RaggedTensorToSparseShapeFn()
230 if (c->RankKnown(encoded_ragged) && input_ragged_rank >= 0) { in RaggedTensorFromVariantShapeFn()
|
| D | parsing_ops.cc | 50 if (c->RankKnown(input_shape)) { in AddSparseOutputShapes()
213 if (c->RankKnown(input) && c->Rank(input) == 1) { in __anonf8f6e66e0502()
375 if (c->RankKnown(input) && c->Rank(input) == 1) { in __anonf8f6e66e0802()
|
| D | tpu_cross_replica_ops.cc | 36 if (c->RankKnown(input)) { in __anonc85edbdb0102()
|
| D | string_ops.cc | 175 if (c->RankKnown(c->input(i)) && c->Rank(c->input(i)) != 0) { in __anon4354d5f10402()
265 if (c->RankKnown(len_shape)) { in __anon4354d5f10702()
|
| D | math_ops.cc | 659 if (!c->RankKnown(in)) { in __anon725c8f640202()
815 if (!c->RankKnown(cond) || !c->RankKnown(data)) { in __anon725c8f640402()
1050 if (!c->RankKnown(input_shape)) { in ArgOpShape()
1595 if (c->RankKnown(a_shape)) { in __anon725c8f640c02()
1940 if (c->RankKnown(input)) { in __anon725c8f641702()
|
| D | lookup_ops.cc | 131 if (c->RankKnown(key_shape_and_type.shape) && c->RankKnown(keys)) { in ValidateTableResourceHandle()
|
| D | list_ops.cc | 274 if (c->RankKnown(element_shape)) { in TensorListConcatShapeInference()
390 if (c->RankKnown((*handle_data)[0].shape)) { in __anonabedecb60b02()
|
| D | data_flow_ops.cc | 61 if (!c->RankKnown(partitions_shape)) { in __anonea01102b0202()
106 if (!c->RankKnown(indices_shape)) { in DynamicStitchShapeFunction()
691 if (c->RankKnown(input_shape) && in __anonea01102b1702()
|
| D | resource_variable_ops.cc | 225 int64 rank = c->RankKnown(var_shape) ? c->Rank(var_shape) in VariableShapeShapeFn()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 230 if (!RankKnown(handle)) { in ShapeHandleToProto()
247 if (!RankKnown(s)) return false; in FullyDefined()
277 if (RankKnown(s)) { in DebugString()
432 if (!RankKnown(prefix) || !RankKnown(s)) { in MergePrefix()
459 } else if (!RankKnown(s_new) || !s_old.IsSet()) { in Relax()
504 } else if (!RankKnown(s1)) { in Merge()
508 } else if (!RankKnown(s0)) { in Merge()
585 ((RankKnown(s) && end >= rank) || in Subshape()
590 if (!RankKnown(s)) { in Subshape()
640 if (!RankKnown(s1) || !RankKnown(s2)) { in Concatenate()
[all …]
|
| D | common_shape_fns.cc | 187 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()
1275 if (!c->RankKnown(input_shape) || !c->FullyDefined(diag_index_shape) || in MatrixDiagPartV2Shape()
1332 if (!c->RankKnown(input_shape) || !c->FullyDefined(diag_index_shape) || in MatrixDiagV2Shape()
1443 if (c->RankKnown(input_shape)) { in MatrixSetDiagV2Shape()
1476 if (c->RankKnown(diag_shape) && !c->FullyDefined(input_shape)) { in MatrixSetDiagV2Shape()
[all …]
|
| D | shape_inference_testutil.cc | 130 if (c.RankKnown(out)) { in InferShapes()
144 if (!c.RankKnown(out)) { in InferShapes()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 34 if (c->RankKnown(c->input(0))) { in UnchangedRank()
420 if (!c->RankKnown(padding_shape_handle) || in __anonbbc1c2d10302()
535 if (c->RankKnown(c->input(0))) { in __anonbbc1c2d10502()
585 if (c->RankKnown(c->input(0))) { in __anonbbc1c2d10702()
846 if (!c->RankKnown(input_handle)) { in __anonbbc1c2d10c02()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 93 if (InferenceContext::RankKnown(*result)) { in Merge()
98 if (InferenceContext::RankKnown(h1)) { in Merge()
368 if (!ic->RankKnown(shape)) { in ReplaceUnknownDimFromConstWithUnknownDim()
650 if (ic->RankKnown(s)) { in StringifyShapeHandle()
1150 if (!ctx->RankKnown(shape2) || ctx->Rank(shape2) != rank) { in OutputAsUnion()
1174 if (!InferenceContext::RankKnown(s1) && !InferenceContext::RankKnown(s2)) { in EquivalentShapes()
1205 if (!InferenceContext::RankKnown(inferred_shape)) { in CompatibleShapes()
1682 if (ic->RankKnown(ic->input(0))) { in MaybeUpdateNodeContextOutput()
1725 if (!ic->RankKnown(input)) { in MaybeUpdateNodeContextOutput()
1759 if (ic->RankKnown(shape_handle) && ic->Rank(shape_handle) >= 1 && in MaybeUpdateNodeContextOutput()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 497 if (!target_context->RankKnown(pre_cast_shape)) { in ConstantPartialShape()
543 if (!target_context->RankKnown(sub_result)) { in ConstantPartialShape()
775 if (!c->RankKnown(s0) && !c->RankKnown(s1)) { in SameDefinedShape()
|
| D | eval_const_tensor.cc | 83 bool rank_known = c->RankKnown(c->input(0)); in TryToInferTensorOutputFromInputShapes()
|
| /external/tensorflow/tensorflow/c/ |
| D | ops.cc | 158 return cc_ctx->RankKnown(*reinterpret_cast<ShapeHandle*>(handle)); in TF_ShapeInferenceContextRankKnown()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 36 if (!context->RankKnown(handle)) return Status::OK(); in ShapeHandleToTensorShape()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 202 if (!context->RankKnown(handle)) return Status::OK(); in ShapeHandleToTensorShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | shape_inference_utils.cc | 146 if (!context.RankKnown(sh)) return None; in GetShapeFromHandle()
|