Home
last modified time | relevance | path

Searched refs:shape_inference (Results 1 – 25 of 184) sorted by relevance

12345678

/external/tensorflow/tensorflow/python/framework/
Dtest_ops.cc27 .SetShapeFn(shape_inference::ScalarShape);
32 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc329d32f0102()
33 shape_inference::ShapeHandle out; in __anonc329d32f0102()
42 .SetShapeFn(shape_inference::ScalarShape);
47 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc329d32f0202()
51 return shape_inference::ScalarShape(c); in __anonc329d32f0202()
55 .SetShapeFn(shape_inference::UnknownShape)
63 .SetShapeFn(shape_inference::ScalarShape);
67 .SetShapeFn(shape_inference::UnknownShape);
71 .SetShapeFn(shape_inference::UnknownShape);
[all …]
/external/tensorflow/tensorflow/core/ops/
Dexperimental_dataset_ops.cc26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0102()
27 shape_inference::ShapeHandle unused; in __anon4afff42d0102()
30 return shape_inference::ScalarShape(c); in __anon4afff42d0102()
39 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0202()
40 shape_inference::ShapeHandle unused; in __anon4afff42d0202()
43 return shape_inference::ScalarShape(c); in __anon4afff42d0202()
52 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0302()
53 shape_inference::ShapeHandle unused; in __anon4afff42d0302()
56 return shape_inference::ScalarShape(c); in __anon4afff42d0302()
68 .SetShapeFn(shape_inference::ScalarShape);
[all …]
Ddataset_ops.cc42 .SetShapeFn(shape_inference::ScalarShape); // TODO(mrry): Validate that
54 .SetShapeFn(shape_inference::ScalarShape); // TODO(mrry): Validate that the
67 .SetShapeFn(shape_inference::ScalarShape);
84 .SetShapeFn(shape_inference::ScalarShape);
92 .SetShapeFn(shape_inference::ScalarShape);
100 .SetShapeFn(shape_inference::ScalarShape);
108 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00102()
109 shape_inference::ShapeHandle count_shape; in __anon42b7b0c00102()
111 return shape_inference::ScalarShape(c); in __anon42b7b0c00102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00202()
[all …]
Drisc_ops.cc25 Status RiscBinaryNonBroadcastOpShapeFn(shape_inference::InferenceContext* c) { in RiscBinaryNonBroadcastOpShapeFn()
52 .SetShapeFn(shape_inference::UnchangedShape);
70 .SetShapeFn(shape_inference::UnchangedShape);
86 .SetShapeFn(shape_inference::UnknownShape);
95 .SetShapeFn(shape_inference::UnknownShape);
102 .SetShapeFn(shape_inference::UnchangedShape);
108 .SetShapeFn(shape_inference::UnchangedShape);
115 .SetShapeFn(shape_inference::UnknownShape);
124 .SetShapeFn(shape_inference::ConcatV2Shape);
136 .SetShapeFn(shape_inference::UnknownShape);
[all …]
Dlist_ops.cc27 shape_inference::InferenceContext* c, in VerifyHandleData()
28 const std::vector<shape_inference::ShapeAndType>& shapes_and_types, in VerifyHandleData()
36 const shape_inference::ShapeAndType& list_shape_type = shapes_and_types[0]; in VerifyHandleData()
47 const std::vector<shape_inference::ShapeAndType>* handle_data) { in IsValidTensorListHandleData()
52 shape_inference::ShapeHandle GetElementShapeFromHandleData( in GetElementShapeFromHandleData()
53 const std::vector<shape_inference::ShapeAndType>& shapes_and_types) { in GetElementShapeFromHandleData()
63 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60202()
67 shape_inference::ShapeHandle element_shape; in __anonabedecb60202()
71 0, std::vector<shape_inference::ShapeAndType>{ in __anonabedecb60202()
81 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60302()
[all …]
Dcollective_ops.cc36 .SetShapeFn(shape_inference::UnchangedShape);
49 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon52c2f9040102()
51 shape_inference::ShapeHandle unused; in __anon52c2f9040102()
54 shape_inference::ShapeHandle in_subshape; in __anon52c2f9040102()
61 shape_inference::ShapeHandle output_first_dim_as_shape; in __anon52c2f9040102()
63 shape_inference::InferenceContext::kUnknownDim) { in __anon52c2f9040102()
65 c->Vector(shape_inference::InferenceContext::kUnknownDim); in __anon52c2f9040102()
69 std::vector<shape_inference::DimensionHandle> output_first_dim; in __anon52c2f9040102()
75 shape_inference::ShapeHandle out; in __anon52c2f9040102()
93 .SetShapeFn(shape_inference::ExplicitShape);
[all …]
Dsummary_ops.cc25 .SetShapeFn(shape_inference::ScalarShape);
33 .SetShapeFn(shape_inference::NoOutputs);
41 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::NoOutputs);
49 .SetShapeFn(shape_inference::NoOutputs);
58 .SetShapeFn(shape_inference::NoOutputs);
64 .SetShapeFn(shape_inference::NoOutputs);
69 .SetShapeFn(shape_inference::NoOutputs);
77 .SetShapeFn(shape_inference::NoOutputs);
85 .SetShapeFn(shape_inference::NoOutputs);
[all …]
Dboosted_trees_ops.cc27 using shape_inference::DimensionHandle;
28 using shape_inference::InferenceContext;
29 using shape_inference::ShapeHandle;
36 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon770f674c0102()
37 shape_inference::ShapeHandle unused_input; in __anon770f674c0102()
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon770f674c0202()
63 shape_inference::ShapeHandle node_id_range_shape; in __anon770f674c0202()
64 shape_inference::ShapeHandle unused_shape; in __anon770f674c0202()
69 shape_inference::ShapeHandle summary_shape_base; in __anon770f674c0202()
75 shape_inference::ShapeHandle summary_shape; in __anon770f674c0202()
[all …]
Dmath_ops.cc23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
52 std::vector<shape_inference::ShapeAndType> cur_shapes_and_types; in __anon725c8f640102()
114 .SetShapeFn(shape_inference::ExplicitShape);
127 .SetShapeFn(shape_inference::BatchMatMulShape);
138 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
148 .SetShapeFn(shape_inference::BatchMatMulShape);
157 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
173 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
Dspectral_ops.cc23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
32 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anona0ecfad80102()
40 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anona0ecfad80202()
48 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anona0ecfad80302()
56 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anona0ecfad80402()
64 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anona0ecfad80502()
72 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anona0ecfad80602()
163 .SetShapeFn(shape_inference::UnknownShape)
[all …]
Ddata_flow_ops.cc23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
40 return shape_inference::UnknownShape(c); in DequeueManyV2Shape()
62 return shape_inference::UnknownShape(c); in __anonea01102b0202()
204 .SetShapeFn(shape_inference::ScalarShape);
224 .SetShapeFn(shape_inference::ScalarShape);
244 .SetShapeFn(shape_inference::ScalarShape);
264 .SetShapeFn(shape_inference::ScalarShape);
277 .SetShapeFn(shape_inference::UnknownShape);
[all …]
Dstateful_random_ops.cc23 Status StatefulRandomShape(shape_inference::InferenceContext* c) { in StatefulRandomShape()
24 using shape_inference::ShapeHandle; in StatefulRandomShape()
59 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20102()
60 using shape_inference::ShapeHandle; in __anon0d38c5e20102()
87 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20202()
88 shape_inference::ShapeHandle unused; in __anon0d38c5e20202()
99 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20302()
100 shape_inference::ShapeHandle unused; in __anon0d38c5e20302()
113 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20402()
114 using shape_inference::ShapeHandle; in __anon0d38c5e20402()
[all …]
Dtensor_forest_ops.cc30 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6270b1470102()
31 shape_inference::ShapeHandle unused_input; in __anon6270b1470102()
40 .SetShapeFn(tensorflow::shape_inference::NoOutputs);
45 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
50 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6270b1470202()
51 shape_inference::ShapeHandle unused_input; in __anon6270b1470202()
59 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
66 .SetShapeFn([](tensorflow::shape_inference::InferenceContext* c) { in __anon6270b1470302()
67 shape_inference::ShapeHandle shape_handle; in __anon6270b1470302()
68 shape_inference::DimensionHandle batch_size = c->UnknownDim(); in __anon6270b1470302()
Dfunctional_ops.cc22 using shape_inference::InferenceContext;
41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anonb31eee6d0102()
63 .SetShapeFn(shape_inference::UnknownShape);
76 .SetShapeFn(shape_inference::UnknownShape)
93 Status IfShapeInferenceFn(shape_inference::InferenceContext* c) { in IfShapeInferenceFn()
98 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in IfShapeInferenceFn()
105 shape_inference::ShapeHandle output_shape_handle; in IfShapeInferenceFn()
138 Status CaseShapeInferenceFn(shape_inference::InferenceContext* c) { in CaseShapeInferenceFn()
143 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in CaseShapeInferenceFn()
150 shape_inference::ShapeHandle output_shape_handle; in CaseShapeInferenceFn()
[all …]
Dlogging_ops.cc23 using shape_inference::InferenceContext;
31 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::UnchangedShape);
79 .SetShapeFn(shape_inference::ScalarShape);
88 .SetShapeFn(shape_inference::ScalarShape);
100 .SetShapeFn(shape_inference::ScalarShape);
108 .SetShapeFn(shape_inference::ScalarShape);
116 .SetShapeFn(shape_inference::ScalarShape)
122 .SetShapeFn(shape_inference::ScalarShape);
Dstring_ops.cc30 namespace shape_inference { namespace
34 using shape_inference::DimensionHandle;
35 using shape_inference::InferenceContext;
36 using shape_inference::ShapeHandle;
58 .SetShapeFn(shape_inference::UnchangedShape);
75 .SetShapeFn(shape_inference::UnchangedShape);
81 .SetShapeFn(shape_inference::UnchangedShape);
88 .SetShapeFn(shape_inference::UnchangedShape);
94 .SetShapeFn(shape_inference::UnchangedShape);
102 .SetShapeFn(shape_inference::ReductionShape);
[all …]
Dnccl_ops.cc21 using shape_inference::InferenceContext;
22 using shape_inference::ShapeHandle;
32 .SetShapeFn(shape_inference::UnchangedShape);
43 .SetShapeFn(shape_inference::UnchangedShape);
52 .SetShapeFn(shape_inference::NoOutputs)
75 .SetShapeFn(shape_inference::UnchangedShape)
100 .SetShapeFn(shape_inference::UnchangedShape);
108 .SetShapeFn(shape_inference::NoOutputs)
Dtpu_infeed_ops.cc22 using shape_inference::InferenceContext;
23 using shape_inference::ShapeHandle;
30 .SetShapeFn(shape_inference::ExplicitShape);
38 .SetShapeFn(shape_inference::NoOutputs)
47 .SetShapeFn(shape_inference::NoOutputs)
72 .SetShapeFn(shape_inference::ScalarShape);
80 .SetShapeFn(shape_inference::ScalarShape);
85 .SetShapeFn(shape_inference::NoOutputs);
Dbatch_ops.cc59 .SetShapeFn(shape_inference::UnknownShape);
76 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond951767c0102()
77 std::vector<shape_inference::ShapeHandle> in_shapes; in __anond951767c0102()
79 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anond951767c0102()
88 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anond951767c0102()
89 shape_inference::DimensionOrConstant(3)})})); in __anond951767c0102()
102 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond951767c0202()
103 shape_inference::ShapeHandle out_shape; in __anond951767c0202()
119 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond951767c0302()
Dnn_ops.cc29 using shape_inference::DimensionHandle;
30 using shape_inference::InferenceContext;
31 using shape_inference::ShapeHandle;
80 .SetShapeFn(shape_inference::AvgPoolShape);
91 .SetShapeFn(shape_inference::AvgPoolGradShape);
181 .SetShapeFn(shape_inference::FusedBatchNormShape);
200 .SetShapeFn(shape_inference::FusedBatchNormShape);
220 .SetShapeFn(shape_inference::FusedBatchNormV3Shape);
243 .SetShapeFn(shape_inference::FusedBatchNormExShape)
266 .SetShapeFn(shape_inference::FusedBatchNormGradShape);
[all …]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h26 namespace shape_inference {
47 Status UnchangedShape(shape_inference::InferenceContext* c);
50 inline Status UnchangedShapeWithRank(shape_inference::InferenceContext* c, in UnchangedShapeWithRank()
60 shape_inference::InferenceContext* c, int32 rank) { in UnchangedShapeWithRankAtLeast()
68 inline Status UnchangedShapeWithRankAtMost(shape_inference::InferenceContext* c, in UnchangedShapeWithRankAtMost()
77 inline Status NoOutputs(shape_inference::InferenceContext* c) { in NoOutputs()
82 inline Status ScalarShape(shape_inference::InferenceContext* c) { in ScalarShape()
96 Status DatasetIteratorShape(shape_inference::InferenceContext* c);
104 shape_inference::InferenceContext* context);
107 Status MatMulShape(shape_inference::InferenceContext* c);
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.h38 std::unique_ptr<shape_inference::InferenceContext> ic, const Node* node) in ExtendedInferenceContext()
53 shape_inference::InferenceContext* get_context() { in get_context()
60 std::unique_ptr<shape_inference::InferenceContext> inference_context_;
100 shape_inference::ShapeHandle shape);
113 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
162 static bool SameDefinedShape(shape_inference::InferenceContext* c,
163 shape_inference::ShapeHandle s0,
164 shape_inference::ShapeHandle s1);
169 shape_inference::InferenceContext* c,
170 const std::vector<shape_inference::ShapeAndType>& existing,
[all …]
/external/tensorflow/tensorflow/compiler/xrt/ops/
Dxrt_state_ops.cc32 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
45 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
63 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
89 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
104 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
121 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
139 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
152 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
165 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
180 .SetShapeFn(tensorflow::shape_inference::UnknownShape)
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc33 Status UnchangedRank(shape_inference::InferenceContext* c) { in UnchangedRank()
50 .SetShapeFn(shape_inference::UnknownShape)
72 .SetShapeFn(shape_inference::UnknownShape)
108 .SetShapeFn(shape_inference::UnknownShape)
171 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonbbc1c2d10202()
172 shape_inference::ShapeHandle lhs_shape_handle = c->input(0); in __anonbbc1c2d10202()
173 shape_inference::ShapeHandle rhs_shape_handle = c->input(1); in __anonbbc1c2d10202()
176 return shape_inference::UnknownShape(c); in __anonbbc1c2d10202()
202 shape_inference::DimensionOrConstant in __anonbbc1c2d10202()
205 shape_inference::DimensionOrConstant in __anonbbc1c2d10202()
[all …]
/external/tensorflow/tensorflow/core/tpu/ops/
Dhost_compute_ops.cc23 using shape_inference::InferenceContext;
24 using shape_inference::ShapeHandle;
34 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc8f95cfc0102()
35 return ::tensorflow::shape_inference::UnknownShape(c); in __anonc8f95cfc0102()
62 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc8f95cfc0202()
75 shape_inference::ShapeHandle handle; in __anonc8f95cfc0202()
84 return ::tensorflow::shape_inference::UnknownShape(c); in __anonc8f95cfc0202()
92 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc8f95cfc0302()
93 return ::tensorflow::shape_inference::UnknownShape(c); in __anonc8f95cfc0302()
103 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc8f95cfc0402()
[all …]

12345678