| /external/tensorflow/tensorflow/core/framework/ |
| D | partial_tensor_shape_test.cc | 113 EXPECT_FALSE(a.IsFullyDefined()); in TEST()
114 EXPECT_FALSE(c.IsFullyDefined()); in TEST()
115 EXPECT_TRUE(b.IsFullyDefined()); in TEST()
116 EXPECT_TRUE(d.IsFullyDefined()); in TEST()
117 EXPECT_TRUE(e.IsFullyDefined()); in TEST()
118 EXPECT_FALSE(f.IsFullyDefined()); in TEST()
264 EXPECT_FALSE(shape.IsFullyDefined()); in TEST()
266 EXPECT_TRUE(shape.IsFullyDefined()); in TEST()
|
| D | tensor_shape.h | 309 bool IsFullyDefined() const { return !kIsPartial || num_elements() != -1; } in IsFullyDefined() function
|
| D | tensor_shape.cc | 927 if (IsFullyDefined()) { in AsTensorShape()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 88 partial_element_shape.IsFullyDefined() || in Compute()
96 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
197 if (!partial_element_shape.IsFullyDefined()) { in Compute()
327 element_shape_except_first_dim.IsFullyDefined(), in Compute()
340 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
569 c, partial_element_shape.IsFullyDefined() || indices.NumElements() > 0, in Compute()
576 if (!tensor_list->element_shape.IsFullyDefined()) { in Compute()
|
| D | tensor_array.h | 479 } else if (identical_element_shapes_ && !element_shape_.IsFullyDefined()) { in LockedWriteOrAggregate()
568 } else if (!element_shape_.IsFullyDefined()) { in LockedRead()
|
| D | padding_fifo_queue.cc | 189 bool has_dynamic_shape = !partial_shape.IsFullyDefined(); in TryDequeueMany()
|
| D | tensor_array_ops.cc | 638 OP_REQUIRES(ctx, element_shape_.IsFullyDefined(), in Compute()
842 ctx, element_shape_except0_.IsFullyDefined(), in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 143 if (!compute_opts_->output_shapes.at(index).IsFullyDefined()) { in SetRetval()
296 if (output_shapes_.at(i).IsFullyDefined()) { in SetupOutputs()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 92 if (!shape.IsFullyDefined()) { in MaybeReplaceShapeOrShapeNOp()
148 if (!input_shapes[0].IsFullyDefined()) { in MaybeReplaceSizeOp()
243 if (out_shape.IsFullyDefined() && in IsConstantFoldable()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 37 TF_RET_CHECK(tshape.IsFullyDefined() && in HostTensorToBorrowingLiteral()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 417 if ((op != "Rank" && !shape.IsFullyDefined()) || in MaterializeShapes()
502 if (!shape.IsFullyDefined()) { in MaterializeShapes()
840 if (!PartialTensorShape(output_shape).IsFullyDefined()) { in MaterializeConstantValuedNode()
1010 if (input_shape.IsFullyDefined()) { in IsFoldableUncached()
1017 if (output_shape.IsFullyDefined()) { in IsFoldableUncached()
1701 if (!shape.IsFullyDefined()) { in IsSimplifiableReshape()
1820 if (!shape.IsFullyDefined()) { in ReplaceOperationWithBroadcastTo()
2997 if (shp.IsFullyDefined() && IsLogicalOr(*node) && (y_is_one || x_is_one)) { in SimplifyArithmeticOperations()
3010 if (shp.IsFullyDefined()) { in SimplifyArithmeticOperations()
3354 if (c_shape.IsFullyDefined() && x_shape.IsFullyDefined() && in ConstantPushDown()
|
| D | memory_optimizer.cc | 608 if (!shp.IsFullyDefined()) { in SchedulingPass()
|
| D | arithmetic_optimizer.cc | 2644 PartialTensorShape(output_shape).IsFullyDefined()) { in TrySimplify()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 204 bool is_fully_defined = TensorShape(shape_proto).IsFullyDefined(); in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 392 if (expected_output_shapes[i].IsFullyDefined()) { in Join()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 534 if (!IsReady() && inference_shape_.IsFullyDefined()) { in Shape()
653 if (!IsReady() && inference_shape_.IsFullyDefined()) { in NumElements()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 458 if (static_shape.IsFullyDefined()) { in ComputeArgumentShapes()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 124 if (!partial_shape.IsFullyDefined()) { in TryGetElementShapeFromInput()
|
| D | tensor_array_ops.cc | 160 if (element_shape_.IsFullyDefined()) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 378 if (shape.IsFullyDefined()) continue; in BuildCompileOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 440 [](PartialTensorShape shape) { return !shape.IsFullyDefined(); }); in TRTEngineOp()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 480 return ishape.shape.IsFullyDefined() in GetInferredShapeSize()
1715 if ((info->handle_type == DT_INVALID && !info->shape.IsFullyDefined()) || in GetArgAndRetvalShapes()
1717 !info->handle_shape.IsFullyDefined())) { in GetArgAndRetvalShapes()
2255 if (!shape->IsFullyDefined()) { in BuildDynamicShapeNodes()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 295 if (!shape.IsFullyDefined()) { in GetInferredInputShapes()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1818 EXPECT_FALSE(shape.IsFullyDefined()); in TEST_F()
|