| /external/tensorflow/tensorflow/core/kernels/ |
| D | padding_fifo_queue.cc | 166 const PartialTensorShape partial_shape = in TryDequeueMany()
171 for (int j = 0; j < partial_shape.dims() - 1; ++j) { in TryDequeueMany()
172 if (partial_shape.dim_size(j + 1) > -1) { in TryDequeueMany()
173 shape.AddDim(partial_shape.dim_size(j + 1)); in TryDequeueMany()
189 bool has_dynamic_shape = !partial_shape.IsFullyDefined(); in TryDequeueMany()
|
| D | functional_ops.cc | 871 PartialTensorShape partial_shape; in FakeParamOp() local
872 OP_REQUIRES_OK(context, context->GetAttr("shape", &partial_shape)); in FakeParamOp()
873 if (!partial_shape.unknown_rank()) { in FakeParamOp()
874 for (int64 d : partial_shape.dim_sizes()) { in FakeParamOp()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_unified_experimental.cc | 148 tensorflow::PartialTensorShape partial_shape; in TF_AddFunctionParameter() local
153 &partial_shape); in TF_AddFunctionParameter()
160 s, tracing_ctx->AddParameter(static_cast<DataType>(dtype), partial_shape, in TF_AddFunctionParameter()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 720 PartialTensorShape partial_shape; in OptimizeGraph()
723 &partial_shape, &type); in OptimizeGraph()
739 if (partial_shape.unknown_rank()) { in OptimizeGraph()
742 for (int i = 0; i < partial_shape.dims(); ++i) { in OptimizeGraph()
743 if (partial_shape.dim_size(i) < 0) { in OptimizeGraph()
744 partial_shape.set_dim(i, 0); in OptimizeGraph()
747 if (!partial_shape.AsTensorShape(&shape)) { in OptimizeGraph()
|
| D | accumulate_n_optimizer.cc | 156 PartialTensorShape partial_shape(shape); in RewriteIntoTempVariable() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 122 PartialTensorShape partial_shape; in TryGetElementShapeFromInput() local
123 TF_RETURN_IF_ERROR(ctx->ConstantInputAsPartialShape(0, &partial_shape)); in TryGetElementShapeFromInput()
124 if (!partial_shape.IsFullyDefined()) { in TryGetElementShapeFromInput()
129 *shape = xla::ShapeUtil::MakeShape(dtype, partial_shape.dim_sizes()); in TryGetElementShapeFromInput()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 863 const PartialTensorShape& partial_shape, ShapeHandle* out) { in MakeShapeFromPartialTensorShape() argument
865 if (partial_shape.dims() == -1) { in MakeShapeFromPartialTensorShape()
868 const int num_dims = partial_shape.dims(); in MakeShapeFromPartialTensorShape()
873 dims[i] = MakeDim(partial_shape.dim_size(i)); in MakeShapeFromPartialTensorShape()
888 PartialTensorShape partial_shape(proto); in MakeShapeFromShapeProto() local
889 return MakeShapeFromPartialTensorShape(partial_shape, out); in MakeShapeFromShapeProto()
|
| D | shape_inference.h | 479 const PartialTensorShape& partial_shape, ShapeHandle* out);
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape_test.py | 422 partial_shape = tensor_shape.TensorShape([None, 37, 42])
423 self.assertNotEqual(partial_proto_shape, partial_shape)
427 self.assertTrue(partial_shape.is_compatible_with(partial_proto_shape))
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2190 const int64_t partial_shape[] = {17, -1}; in TEST_F() local
2191 const size_t sz = TF_ARRAYSIZE(partial_shape); in TEST_F()
2193 TF_SetAttrShape(desc, "v", partial_shape, sz); in TEST_F()
2201 std::equal(std::begin(partial_shape), std::end(partial_shape), values)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 6473 PartialTensorShape partial_shape; in ConvertSegmentToGraphDef() local
6477 connection.outside_port, &partial_shape, &dtype); in ConvertSegmentToGraphDef()
6478 connection.outside_shape = partial_shape; in ConvertSegmentToGraphDef()
6482 connection.outside_port, &partial_shape, &dtype); in ConvertSegmentToGraphDef()
6483 connection.inside_shape = partial_shape; in ConvertSegmentToGraphDef()
6501 auto status = builder.Attr("shape", partial_shape) in ConvertSegmentToGraphDef()
|
| D | convert_nodes_test.cc | 1831 std::vector<int32> partial_shape; local
1833 partial_shape = partial_input_shape_dims;
1837 partial_shape = std::vector<int32>(dims.size(), -1);
1840 partial_shape = dims;
1845 AddTestTensorWithTFDims(name, partial_shape, trt_type, add_input_status);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 874 const Shape& partial_shape = rhs->operand(1)->shape(); in HandleAdd() local
877 partial_shape, lhs, absl::MakeSpan(rhs->operands()).subspan(2), in HandleAdd()
878 partial_shape.dimensions())); in HandleAdd()
|