Home
last modified time | relevance | path

Searched refs:IsFullyDefined (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dpartial_tensor_shape_test.cc100 EXPECT_FALSE(a.IsFullyDefined()); in TEST()
101 EXPECT_FALSE(c.IsFullyDefined()); in TEST()
102 EXPECT_TRUE(b.IsFullyDefined()); in TEST()
103 EXPECT_TRUE(d.IsFullyDefined()); in TEST()
104 EXPECT_TRUE(e.IsFullyDefined()); in TEST()
105 EXPECT_FALSE(f.IsFullyDefined()); in TEST()
251 EXPECT_FALSE(shape.IsFullyDefined()); in TEST()
253 EXPECT_TRUE(shape.IsFullyDefined()); in TEST()
Dtensor_shape.h248 bool IsFullyDefined() const { return !kIsPartial || num_elements() != -1; } in IsFullyDefined() function
Dtensor_shape.cc682 if (IsFullyDefined()) { in AsTensorShape()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h110 partial_element_shape.IsFullyDefined() || !tensor_list->tensors.empty(), in Compute()
117 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
218 if (!partial_element_shape.IsFullyDefined()) { in Compute()
351 element_shape_except_first_dim_.IsFullyDefined(), in Compute()
364 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
592 c, partial_element_shape.IsFullyDefined() || indices.NumElements() > 0, in Compute()
599 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
Dtensor_array.h480 } else if (identical_element_shapes_ && !element_shape_.IsFullyDefined()) { in LockedWriteOrAggregate()
569 } else if (!element_shape_.IsFullyDefined()) { in LockedRead()
Dpadding_fifo_queue.cc189 bool has_dynamic_shape = !partial_shape.IsFullyDefined(); in TryDequeueMany()
Dtensor_array_ops.cc643 OP_REQUIRES(ctx, element_shape_.IsFullyDefined(), in Compute()
848 ctx, element_shape_except0_.IsFullyDefined(), in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc228 if (!compute_opts_->output_shapes.at(index).IsFullyDefined()) { in SetRetval()
306 if (output_shapes_.at(i).IsFullyDefined()) { in SetupOutputs()
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding.cc90 if (!shape.IsFullyDefined()) { in MaybeReplaceShapeOrShapeNOp()
146 if (!input_shapes[0].IsFullyDefined()) { in MaybeReplaceSizeOp()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc204 bool is_fully_defined = TensorShape(shape_proto).IsFullyDefined(); in ReplaceBroadcast()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc78 partial_element_shape.IsFullyDefined()) { in CreateZerosList()
Dtensor_array_ops.cc160 if (element_shape_.IsFullyDefined()) { in Compile()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc374 if ((op != "Rank" && !shape.IsFullyDefined()) || in MaterializeShapes()
450 if (!shape.IsFullyDefined()) { in MaterializeShapes()
782 if (!PartialTensorShape(output_shape).IsFullyDefined()) { in MaterializeConstantValuedNode()
1531 if (!shape.IsFullyDefined()) { in IsSimplifiableReshape()
2712 if (shp.IsFullyDefined() && IsLogicalOr(*node) && (y_is_one || x_is_one)) { in SimplifyArithmeticOperations()
2730 if (shp.IsFullyDefined()) { in SimplifyArithmeticOperations()
Dmemory_optimizer.cc587 if (!shp.IsFullyDefined()) { in SchedulingPass()
Darithmetic_optimizer.cc2474 if (!shape.IsFullyDefined()) { in TrySimplify()
2525 if (!shape.IsFullyDefined()) { in TrySimplify()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc281 if (!shape.IsFullyDefined()) { in GetInferredInputShapes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1576 EXPECT_FALSE(shape.IsFullyDefined()); in TEST_F()