Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc56 if (c->RankKnown(input0_shape)) { in __anon967df2c80102()
61 if (c->RankKnown(input1_shape)) { in __anon967df2c80102()
78 if (c->RankKnown(input1_shape)) { in __anon967df2c80102()
118 if (c->RankKnown(input0_shape)) { in __anon967df2c80202()
Darray_ops.cc86 } else if (c->RankKnown(input)) { in PadShapeFn()
122 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()
129 if (c->RankKnown(input)) { in TransposeShapeFn()
136 if (!c->RankKnown(input) && rank < 2) { in TransposeShapeFn()
185 if (!c->RankKnown(out)) { in SetOutputShapeForReshape()
191 if (c->RankKnown(out) && c->RankKnown(in)) { in SetOutputShapeForReshape()
331 if (!c->RankKnown(cur)) { in __anon7c94107b0302()
417 if (c->RankKnown(s)) { in __anon7c94107b0502()
452 if (c->RankKnown(indices) && c->Rank(indices) == 0) { in __anon7c94107b0602()
454 } else if (c->RankKnown(indices)) { in __anon7c94107b0602()
[all …]
Dcontrol_flow_ops.cc93 if (!c->RankKnown(out)) { in MergeShape()
99 if (!c->RankKnown(input) || c->Rank(input) != rank) { in MergeShape()
Dragged_conversion_ops.cc62 c->RankKnown(rt_dense_values) in RaggedTensorToSparseShapeFn()
Dtpu_cross_replica_ops.cc36 if (c->RankKnown(input)) { in __anonb72bce3d0102()
Drpc_ops.cc41 if (!c->RankKnown(output)) { in RpcShapeOp()
Dmath_ops.cc631 if (!c->RankKnown(in)) { in __anonb22bfa860302()
778 if (!c->RankKnown(cond) || !c->RankKnown(data)) { in __anonb22bfa860502()
913 if (!c->RankKnown(input_shape)) { in ArgOpShape()
1112 if (c->RankKnown(s_segment_ids)) { in UnsortedSegmentReductionShapeFn()
1469 if (c->RankKnown(a_shape)) { in __anonb22bfa860c02()
1694 if (c->RankKnown(input)) { in __anonb22bfa861402()
Dlookup_ops.cc123 if (c->RankKnown(key_shape_and_type.shape) && c->RankKnown(keys)) { in ValidateTableResourceHandle()
Dstring_ops.cc165 if (c->RankKnown(c->input(i)) && c->Rank(c->input(i)) != 0) { in __anonb4ab91930402()
Ddata_flow_ops.cc61 if (!c->RankKnown(partitions_shape)) { in __anon4a7ac08d0202()
106 if (!c->RankKnown(indices_shape)) { in DynamicStitchShapeFunction()
636 if (c->RankKnown(input_shape) && in __anon4a7ac08d1302()
Dresource_variable_ops.cc242 int64 rank = c->RankKnown(var_shape) ? c->Rank(var_shape) in VariableShapeShapeFn()
Dlist_ops.cc263 if (c->RankKnown(element_shape)) { in TensorListConcatShapeInference()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc287 if (!RankKnown(handle)) { in ShapeHandleToProto()
304 if (!RankKnown(s)) return false; in FullyDefined()
334 if (RankKnown(s)) { in DebugString()
490 if (!RankKnown(prefix) || !RankKnown(s)) { in MergePrefix()
517 } else if (!RankKnown(s_new) || !s_old.IsSet()) { in Relax()
562 } else if (!RankKnown(s1)) { in Merge()
566 } else if (!RankKnown(s0)) { in Merge()
643 ((RankKnown(s) && end >= rank) || in Subshape()
648 if (!RankKnown(s)) { in Subshape()
698 if (!RankKnown(s1) || !RankKnown(s2)) { in Concatenate()
[all …]
Dshape_inference_testutil.cc131 if (c.RankKnown(out)) { in InferShapes()
146 if (!c.RankKnown(out)) { in InferShapes()
Dcommon_shape_fns.cc254 if (!c->RankKnown(input_shape)) { in BiasAddShape()
1170 if (reduction_indices_t == nullptr || !c->RankKnown(input)) { in ReductionShape()
1174 if (keep_dims && c->RankKnown(input)) { in ReductionShape()
1317 if (!c->RankKnown(shape_x) || !c->RankKnown(shape_y)) { in BroadcastBinaryOpOutputShapeFnHelper()
1471 if (c->RankKnown(sizes_value)) { in SliceShape()
1483 if (c->RankKnown(input)) { in SliceShape()
1556 if (c->RankKnown(indices_shape) && c->RankKnown(updates_shape)) { in ScatterNdUpdateShape()
Dshape_inference.h361 static bool RankKnown(ShapeHandle s) { in RankKnown() function
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc30 if (c->RankKnown(c->input(0))) { in UnchangedRank()
319 if (c->RankKnown(c->input(0))) { in __anonc06300730302()
522 if (context->RankKnown(input_a)) { in __anonc06300730602()
527 if (context->RankKnown(input_b)) { in __anonc06300730602()
/external/tensorflow/tensorflow/contrib/tensor_forest/ops/
Dmodel_ops.cc97 if (c->RankKnown(c->input(1)) && c->Rank(c->input(1)) > 0 && in __anon5b1697710102()
132 if (c->RankKnown(c->input(1)) && c->Rank(c->input(1)) > 0 && in __anon5b1697710202()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc86 if (InferenceContext::RankKnown(*result)) { in Merge()
91 if (InferenceContext::RankKnown(h1)) { in Merge()
938 if (!ctx->RankKnown(shape2) || ctx->Rank(shape2) != rank) { in OutputAsUnion()
962 if (!InferenceContext::RankKnown(s1) && !InferenceContext::RankKnown(s2)) { in EquivalentShapes()
993 if (!InferenceContext::RankKnown(inferred_shape)) { in CompatibleShapes()
1387 if (ic->RankKnown(ic->input(0))) { in MaybeUpdateNodeContextOutput()
1430 if (!ic->RankKnown(input)) { in MaybeUpdateNodeContextOutput()
1462 if (ic->RankKnown(shape_handle) && ic->Rank(shape_handle) >= 1 && in MaybeUpdateNodeContextOutput()
1483 bool valid = ic->RankKnown(input); in MaybeUpdateNodeContextOutput()
1507 bool valid = ic->RankKnown(input); in MaybeUpdateNodeContextOutput()
[all …]
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
Dreduce_slice_ops.cc37 if (c->RankKnown(c->input(1))) { in ReduceSliceShapeFn()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc525 if (!target_context->RankKnown(sub_result)) { in ConstantPartialShape()
732 if (!c->RankKnown(s0) && !c->RankKnown(s1)) { in SameDefinedShape()
Deval_const_tensor.cc79 bool rank_known = c->RankKnown(c->input(0)); in TryToInferTensorOutputFromInputShapes()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc33 if (!context->RankKnown(handle)) return Status::OK(); in ShapeHandleToTensorShape()
/external/tensorflow/tensorflow/python/framework/
Dcpp_shape_inference.cc35 if (c->RankKnown(s)) { in ProtoFromShapeHandle()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc202 if (!context->RankKnown(handle)) return Status::OK(); in ShapeHandleToTensorShape()

12