Searched refs:partial_shape (Results 1 – 9 of 9) sorted by relevance
| /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 | 640 PartialTensorShape partial_shape; in FakeParamOp() local
641 OP_REQUIRES_OK(context, context->GetAttr("shape", &partial_shape)); in FakeParamOp()
642 if (!partial_shape.unknown_rank()) { in FakeParamOp()
643 for (int64 d : partial_shape.dim_sizes()) { in FakeParamOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 655 PartialTensorShape partial_shape; in OptimizeGraph() local
658 GetFeedShapeAndTypeFromAttribute(node, &partial_shape, &type)); in OptimizeGraph()
664 if (partial_shape.unknown_rank()) { in OptimizeGraph()
667 for (int i = 0; i < partial_shape.dims(); ++i) { in OptimizeGraph()
668 if (partial_shape.dim_size(i) < 0) { in OptimizeGraph()
669 partial_shape.set_dim(i, 0); in OptimizeGraph()
672 if (!partial_shape.AsTensorShape(&shape)) { in OptimizeGraph()
|
| D | accumulate_n_optimizer.cc | 124 PartialTensorShape partial_shape(shape); in rewriteNode() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 919 const PartialTensorShape& partial_shape, ShapeHandle* out) { in MakeShapeFromPartialTensorShape() argument
921 if (partial_shape.dims() == -1) { in MakeShapeFromPartialTensorShape()
924 const int num_dims = partial_shape.dims(); in MakeShapeFromPartialTensorShape()
929 dims[i] = MakeDim(partial_shape.dim_size(i)); in MakeShapeFromPartialTensorShape()
944 PartialTensorShape partial_shape(proto); in MakeShapeFromShapeProto() local
945 return MakeShapeFromPartialTensorShape(partial_shape, out); in MakeShapeFromShapeProto()
|
| D | shape_inference.h | 493 const PartialTensorShape& partial_shape, ShapeHandle* out);
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape_test.py | 378 partial_shape = tensor_shape.TensorShape([None, 37, 42])
379 self.assertNotEqual(partial_proto_shape, partial_shape)
383 self.assertTrue(partial_shape.is_compatible_with(partial_proto_shape))
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2152 const int64_t partial_shape[] = {17, -1}; in TEST_F() local
2153 const size_t sz = TF_ARRAYSIZE(partial_shape); in TEST_F()
2155 TF_SetAttrShape(desc, "v", partial_shape, sz); in TEST_F()
2163 std::equal(std::begin(partial_shape), std::end(partial_shape), values)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4187 PartialTensorShape partial_shape; in ConvertSegmentToGraphDef() local
4191 connection.outside_port, &partial_shape, &dtype); in ConvertSegmentToGraphDef()
4192 connection.outside_shape = partial_shape; in ConvertSegmentToGraphDef()
4196 connection.outside_port, &partial_shape, &dtype); in ConvertSegmentToGraphDef()
4197 connection.inside_shape = partial_shape; in ConvertSegmentToGraphDef()
4214 auto status = builder.Attr("shape", partial_shape) in ConvertSegmentToGraphDef()
|