Home
last modified time | relevance | path

Searched refs:PartialTensorShape (Results 1 – 25 of 198) sorted by relevance

12345678

/external/tensorflow/tensorflow/core/framework/
Dpartial_tensor_shape_test.cc29 const PartialTensorShape s; in TEST()
35 const PartialTensorShape s({10, 5}); in TEST()
63 const auto s4 = s.Concatenate(PartialTensorShape()); in TEST()
70 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
74 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
79 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
83 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
86 EXPECT_FALSE(PartialTensorShape::IsValid(proto)); in TEST()
90 EXPECT_FALSE(PartialTensorShape::IsValid(proto)); in TEST()
94 const PartialTensorShape a({-1, 0, 1}); in TEST()
[all …]
Dtensor_shape.h38 class PartialTensorShape; variable
263 std::is_same<Shape, PartialTensorShape>::value;
295 operator const PartialTensorShape&() const; // NOLINT(runtime/explicit)
380 static Status MakeShape(const int32* dims, int64 n, PartialTensorShape* out);
381 static Status MakeShape(const int64* dims, int64 n, PartialTensorShape* out);
383 PartialTensorShape* out);
385 PartialTensorShape* out);
402 class PartialTensorShape : public TensorShapeBase<PartialTensorShape> {
404 PartialTensorShape() {} in PartialTensorShape() function
405 using TensorShapeBase<PartialTensorShape>::TensorShapeBase;
[all …]
Dtensor_shape.cc32 static_assert(sizeof(TensorShapeRep) == sizeof(PartialTensorShape),
511 const auto& shape = *static_cast<const PartialTensorShape*>(this); in DebugString()
618 MAKE_SHAPE(int32, PartialTensorShape) in MAKE_SHAPE()
619 MAKE_SHAPE(int64, PartialTensorShape) in MAKE_SHAPE()
634 PartialTensorShape PartialTensorShape::Concatenate(int64 size) const { in Concatenate()
635 PartialTensorShape out = *this; in Concatenate()
640 PartialTensorShape PartialTensorShape::Concatenate( in Concatenate()
641 const PartialTensorShape& shape) const { in Concatenate()
643 return PartialTensorShape(); in Concatenate()
645 PartialTensorShape out = *this; in Concatenate()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc53 std::map<string, std::vector<PartialTensorShape>> expected = { in TEST()
54 {"A", {PartialTensorShape({2, 3})}}, {"B", {PartialTensorShape({3})}}, in TEST()
55 {"C", {PartialTensorShape()}}, {"D", {PartialTensorShape({2, 3})}}, in TEST()
56 {"E", {PartialTensorShape()}}, {"F", {PartialTensorShape()}}, in TEST()
57 {"G", {PartialTensorShape()}}, in TEST()
116 std::map<string, std::vector<PartialTensorShape>> expected = { in TEST()
117 {"while/Identity", {PartialTensorShape()}}, in TEST()
118 {"while/add", {PartialTensorShape({})}}, in TEST()
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc41 DataTypeVector dtypes, std::vector<PartialTensorShape> shapes, in CreateTensorSliceDatasetKernel()
141 std::vector<PartialTensorShape> shapes; in TEST_F()
211 std::vector<PartialTensorShape> shapes = {PartialTensorShape({2}), in TEST_F()
212 PartialTensorShape({2})}; in TEST_F()
240 std::vector<PartialTensorShape> shapes; in TEST_F()
283 std::vector<PartialTensorShape> shapes; in TEST_F()
305 const std::vector<PartialTensorShape> produced_output_shapes = in TEST_F()
307 std::vector<PartialTensorShape> expected_output_shapes; in TEST_F()
327 std::vector<PartialTensorShape> shapes; in TEST_F()
363 std::vector<PartialTensorShape> shapes = {PartialTensorShape({2}), in TEST_F()
[all …]
Dconcatenate_dataset_op.cc74 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
190 static PartialTensorShape MostSpecificCompatibleShape( in MostSpecificCompatibleShape()
191 const PartialTensorShape& ts1, const PartialTensorShape& ts2) { in MostSpecificCompatibleShape()
193 return PartialTensorShape(); in MostSpecificCompatibleShape()
194 PartialTensorShape output_tensorshape({}); in MostSpecificCompatibleShape()
208 std::vector<PartialTensorShape> output_shapes_;
Dpadded_batch_dataset_op.cc52 std::vector<PartialTensorShape> padded_shapes; in MakeDataset()
64 PartialTensorShape padded_shape; in MakeDataset()
65 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
104 std::vector<PartialTensorShape> padded_shapes, in Dataset()
126 PartialTensorShape({batch_size_}).Concatenate(padded_shapes_[i])); in Dataset()
129 PartialTensorShape({-1}).Concatenate(padded_shapes_[i])); in Dataset()
146 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
275 const PartialTensorShape& padded_shape = in GetNextInternal()
384 const std::vector<PartialTensorShape> padded_shapes_;
387 std::vector<PartialTensorShape> output_shapes_;
Dwindow_dataset.cc26 std::vector<PartialTensorShape> output_shapes) in WindowDataset()
40 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
105 const std::vector<PartialTensorShape> output_shapes_;
112 std::vector<PartialTensorShape> output_shapes, in NewWindowDataset()
Dgraph_rewrite_dataset.h38 const std::vector<PartialTensorShape>& output_shapes) in GraphRewriteDataset()
58 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
89 const std::vector<PartialTensorShape> output_shapes_;
Diterator_ops.h73 std::vector<PartialTensorShape> output_shapes_;
97 std::vector<PartialTensorShape> output_shapes_;
135 std::vector<PartialTensorShape> output_shapes_;
161 std::vector<PartialTensorShape> output_shapes_;
Dfilter_by_component_dataset_op.cc49 std::vector<PartialTensorShape> output_shapes_;
55 std::vector<PartialTensorShape> output_shapes) in Dataset()
74 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
98 const std::vector<PartialTensorShape> output_shapes_;
Dconcatenate_dataset_op_test.cc49 const std::vector<PartialTensorShape> &output_shapes, in CreateConcatenateDatasetKernel()
76 std::vector<PartialTensorShape> expected_output_shapes;
102 {PartialTensorShape({2}), PartialTensorShape({2})}, in TestCase1()
129 {PartialTensorShape({-1}), PartialTensorShape({-1})}, in TestCase2()
214 /*expected_output_shapes*/ {PartialTensorShape({2})}, in TEST_F()
Drange_dataset_op.cc65 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
66 static std::vector<PartialTensorShape>* shapes = in output_shapes()
67 new std::vector<PartialTensorShape>({PartialTensorShape({})}); in output_shapes()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.h45 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
48 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
56 const Tensor& strides_tensor, const PartialTensorShape& input_shape,
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc52 PartialTensorShape row_shape; in MakeDataset()
53 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
82 const PartialTensorShape& row_shape, const DatasetBase* input) in Dataset()
90 PartialTensorShape output_shape({-1}); in Dataset()
108 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
165 const PartialTensorShape& row_shape = in GetNextInternal()
309 const PartialTensorShape row_shape_;
311 std::vector<PartialTensorShape> output_shapes_;
Ddirected_interleave_dataset_op.cc43 PartialTensorShape({})), in MakeDataset()
104 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
258 static PartialTensorShape MostSpecificCompatibleShape( in MostSpecificCompatibleShape()
259 const PartialTensorShape& ts1, const PartialTensorShape& ts2) { in MostSpecificCompatibleShape()
260 PartialTensorShape output_tensorshape; in MostSpecificCompatibleShape()
276 std::vector<PartialTensorShape> output_shapes_;
Dnon_serializable_dataset_op.cc44 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
63 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
119 const std::vector<PartialTensorShape> output_shapes_;
123 std::vector<PartialTensorShape> output_shapes_;
Dindexed_dataset_op.cc54 virtual const std::vector<PartialTensorShape>& output_shapes() const = 0;
105 const std::vector<PartialTensorShape>& output_shapes) in MaterializedDatasetResource()
128 const std::vector<PartialTensorShape>& output_shapes() const { in output_shapes()
147 const std::vector<PartialTensorShape> output_shapes_;
322 std::vector<PartialTensorShape> output_shapes_;
435 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
436 static std::vector<PartialTensorShape>* shapes = in output_shapes()
437 new std::vector<PartialTensorShape>({{}}); in output_shapes()
505 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
Dsql_dataset_op.cc48 for (const PartialTensorShape& pts : output_shapes_) { in SqlDatasetOp()
84 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
102 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
211 const std::vector<PartialTensorShape> output_shapes_;
214 std::vector<PartialTensorShape> output_shapes_;
Dassert_next_dataset_op.cc51 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
71 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
151 const std::vector<PartialTensorShape> output_shapes_;
155 std::vector<PartialTensorShape> output_shapes_;
/external/tensorflow/tensorflow/core/kernels/
Dpadding_fifo_queue.h38 const std::vector<PartialTensorShape>& component_shapes,
60 const gtl::ArraySlice<PartialTensorShape>& partial_shapes);
72 std::vector<PartialTensorShape> partial_shapes_;
Dpadding_fifo_queue.cc38 const std::vector<PartialTensorShape>& partial_shapes, const string& name) in PaddingFIFOQueue()
166 const PartialTensorShape partial_shape = in TryDequeueMany()
167 PartialTensorShape({batch_size}) in TryDequeueMany()
256 const PartialTensorShape expected_shape = in ValidateManyTuple()
257 PartialTensorShape({batch_size}).Concatenate(partial_shapes_[i]); in ValidateManyTuple()
270 std::vector<PartialTensorShape> requested_shapes; in CompatibleNodeDefShapes()
399 const gtl::ArraySlice<PartialTensorShape>& partial_shapes) { in ConvertShapesPartialDimensionsToZero()
402 const PartialTensorShape& partial = partial_shapes[i]; in ConvertShapesPartialDimensionsToZero()
Dlist_kernels.h60 PartialTensorShape element_shape;
67 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out);
71 PartialTensorShape* element_shape);
105 PartialTensorShape partial_element_shape; in Compute()
121 PartialTensorShape tmp = partial_element_shape; in Compute()
205 PartialTensorShape partial_element_shape; in Compute()
221 PartialTensorShape tmp = partial_element_shape; in Compute()
270 PartialTensorShape partial_element_shape; in Compute()
309 PartialTensorShape element_shape; in TensorListConcat()
312 element_shape_except_first_dim_ = PartialTensorShape( in TensorListConcat()
[all …]
/external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/
Dignite_dataset.h30 std::vector<PartialTensorShape> shapes);
35 const std::vector<PartialTensorShape>& output_shapes() const override;
58 const std::vector<PartialTensorShape> shapes_;
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_sample_keys_dataset_op.cc56 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes()
57 static std::vector<PartialTensorShape>* shapes = in output_shapes()
58 new std::vector<PartialTensorShape>({{}}); in output_shapes()

12345678