| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 27 .SetShapeFn(shape_inference::ScalarShape);
32 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonf2f290530102()
33 shape_inference::ShapeHandle out; in __anonf2f290530102()
42 .SetShapeFn(shape_inference::ScalarShape);
47 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonf2f290530202()
51 return shape_inference::ScalarShape(c); in __anonf2f290530202()
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/ |
| D | experimental_dataset_ops.cc | 26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510102()
27 shape_inference::ShapeHandle unused; in __anon03b595510102()
30 return shape_inference::ScalarShape(c); in __anon03b595510102()
39 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510202()
40 shape_inference::ShapeHandle unused; in __anon03b595510202()
43 return shape_inference::ScalarShape(c); in __anon03b595510202()
54 .SetShapeFn(shape_inference::ScalarShape);
64 .SetShapeFn(shape_inference::ScalarShape);
72 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510302()
73 shape_inference::ShapeHandle tag_shape; in __anon03b595510302()
[all …]
|
| D | dataset_ops.cc | 42 .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 __anonc0e8a7640102()
109 shape_inference::ShapeHandle count_shape; in __anonc0e8a7640102()
111 return shape_inference::ScalarShape(c); in __anonc0e8a7640102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640202()
[all …]
|
| D | list_ops.cc | 26 shape_inference::InferenceContext* c, in VerifyHandleData()
27 const std::vector<shape_inference::ShapeAndType>& shapes_and_types, in VerifyHandleData()
35 const shape_inference::ShapeAndType& list_shape_type = shapes_and_types[0]; in VerifyHandleData()
46 shape_inference::ShapeHandle GetElementShapeFromHandleData( in GetElementShapeFromHandleData()
47 const std::vector<shape_inference::ShapeAndType>& shapes_and_types) { in GetElementShapeFromHandleData()
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0202()
61 shape_inference::ShapeHandle element_shape; in __anonef6dc5da0202()
65 0, std::vector<shape_inference::ShapeAndType>{ in __anonef6dc5da0202()
75 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0302()
79 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonef6dc5da0302()
[all …]
|
| D | boosted_trees_ops.cc | 27 using shape_inference::DimensionHandle;
28 using shape_inference::InferenceContext;
29 using shape_inference::ShapeHandle;
36 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9fefc8f00102()
37 shape_inference::ShapeHandle unused_input; in __anon9fefc8f00102()
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9fefc8f00202()
63 shape_inference::ShapeHandle node_id_range_shape; in __anon9fefc8f00202()
64 shape_inference::ShapeHandle unused_shape; in __anon9fefc8f00202()
69 shape_inference::ShapeHandle summary_shape_base; in __anon9fefc8f00202()
75 shape_inference::ShapeHandle summary_shape; in __anon9fefc8f00202()
[all …]
|
| D | summary_ops.cc | 25 .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 …]
|
| D | math_ops.cc | 23 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 __anonb5dc68880102()
114 .SetShapeFn(shape_inference::ExplicitShape);
127 .SetShapeFn(shape_inference::BatchMatMulShape);
138 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
150 .SetShapeFn(shape_inference::BatchMatMulShape);
161 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
177 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
|
| D | logging_ops.cc | 23 using shape_inference::InferenceContext;
31 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::UnchangedShape);
79 .SetShapeFn(shape_inference::ScalarShape);
88 .SetShapeFn(shape_inference::ScalarShape);
95 .SetShapeFn(shape_inference::ScalarShape);
102 .SetShapeFn(shape_inference::ScalarShape);
114 .SetShapeFn(shape_inference::ScalarShape);
122 .SetShapeFn(shape_inference::ScalarShape);
130 .SetShapeFn(shape_inference::ScalarShape)
[all …]
|
| D | spectral_ops.cc | 23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
32 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anone53cc5fc0102()
40 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anone53cc5fc0202()
48 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anone53cc5fc0302()
56 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anone53cc5fc0402()
64 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anone53cc5fc0502()
72 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anone53cc5fc0602()
163 .SetShapeFn(shape_inference::UnknownShape)
[all …]
|
| D | data_flow_ops.cc | 23 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 __anonb84a3fcf0202()
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 …]
|
| D | collective_ops.cc | 35 .SetShapeFn(shape_inference::UnchangedShape);
47 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonea321d280102()
49 shape_inference::ShapeHandle unused; in __anonea321d280102()
52 shape_inference::ShapeHandle in_subshape; in __anonea321d280102()
59 shape_inference::ShapeHandle output_first_dim_as_shape; in __anonea321d280102()
61 shape_inference::InferenceContext::kUnknownDim) { in __anonea321d280102()
63 c->Vector(shape_inference::InferenceContext::kUnknownDim); in __anonea321d280102()
67 std::vector<shape_inference::DimensionHandle> output_first_dim; in __anonea321d280102()
73 shape_inference::ShapeHandle out; in __anonea321d280102()
90 .SetShapeFn(shape_inference::ExplicitShape);
[all …]
|
| D | tensor_forest_ops.cc | 30 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8b5112eb0102()
31 shape_inference::ShapeHandle unused_input; in __anon8b5112eb0102()
40 .SetShapeFn(tensorflow::shape_inference::NoOutputs);
45 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
50 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8b5112eb0202()
51 shape_inference::ShapeHandle unused_input; in __anon8b5112eb0202()
59 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
66 .SetShapeFn([](tensorflow::shape_inference::InferenceContext* c) { in __anon8b5112eb0302()
67 shape_inference::ShapeHandle shape_handle; in __anon8b5112eb0302()
68 shape_inference::DimensionHandle batch_size = c->UnknownDim(); in __anon8b5112eb0302()
|
| D | state_ops.cc | 21 using shape_inference::InferenceContext;
22 using shape_inference::ShapeHandle;
31 .SetShapeFn(shape_inference::ExplicitShape);
48 return shape_inference::UnknownShape(c); in __anon5d8c585f0102()
62 .SetShapeFn(shape_inference::ScalarShape);
70 .SetShapeFn(shape_inference::ExplicitShape);
77 .SetShapeFn(shape_inference::UnchangedShape);
91 return shape_inference::MergeBothInputsShapeFn(c); in __anon5d8c585f0202()
104 .SetShapeFn(shape_inference::MergeBothInputsShapeFn);
112 .SetShapeFn(shape_inference::MergeBothInputsShapeFn);
[all …]
|
| D | stateful_random_ops.cc | 22 Status StatefulRandomShape(shape_inference::InferenceContext* c) { in StatefulRandomShape()
23 using shape_inference::ShapeHandle; in StatefulRandomShape()
58 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860102()
59 using shape_inference::ShapeHandle; in __anonefb820860102()
76 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860202()
77 shape_inference::ShapeHandle unused; in __anonefb820860202()
89 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860302()
90 using shape_inference::ShapeHandle; in __anonefb820860302()
107 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860402()
108 using shape_inference::ShapeHandle; in __anonefb820860402()
[all …]
|
| D | functional_ops.cc | 22 using shape_inference::InferenceContext;
41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anon4a8e12910102()
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()
147 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4a8e12910202()
152 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in __anon4a8e12910202()
159 shape_inference::ShapeHandle output_shape_handle; in __anon4a8e12910202()
[all …]
|
| D | string_ops.cc | 30 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 …]
|
| D | nccl_ops.cc | 21 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)
|
| D | tpu_infeed_ops.cc | 22 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);
|
| D | batch_ops.cc | 40 .SetShapeFn(shape_inference::UnknownShape);
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8ccc74200102()
58 std::vector<shape_inference::ShapeHandle> in_shapes; in __anon8ccc74200102()
60 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anon8ccc74200102()
69 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon8ccc74200102()
70 shape_inference::DimensionOrConstant(3)})})); in __anon8ccc74200102()
83 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8ccc74200202()
84 shape_inference::ShapeHandle out_shape; in __anon8ccc74200202()
100 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8ccc74200302()
|
| D | debug_ops.cc | 35 .SetShapeFn(shape_inference::UnchangedShape);
44 .SetShapeFn(shape_inference::UnchangedShape);
55 .SetShapeFn(shape_inference::UnchangedShape);
66 .SetShapeFn(shape_inference::ScalarShape);
81 .SetShapeFn(shape_inference::UnknownShape);
94 .SetShapeFn(shape_inference::UnchangedShape);
103 .SetShapeFn(shape_inference::UnknownShape);
|
| D | function_ops.cc | 30 .SetShapeFn([](shape_inference::InferenceContext* context) { in __anona02766840102()
53 shape_inference::ShapeHandle shape_handle; in __anona02766840102()
58 0, std::vector<shape_inference::ShapeAndType>{ in __anona02766840102()
72 shape_inference::ShapeHandle shape_handle; in __anona02766840102()
107 .SetShapeFn([](shape_inference::InferenceContext* context) { in __anona02766840202()
123 .SetShapeFn([](shape_inference::InferenceContext* context) { in __anona02766840302()
138 .SetShapeFn([](shape_inference::InferenceContext* context) { in __anona02766840402()
154 .SetShapeFn(shape_inference::UnknownShape)
165 .SetShapeFn(shape_inference::UnknownShape)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 149 namespace shape_inference {
170 Status UnchangedShape(shape_inference::InferenceContext* c);
173 inline Status UnchangedShapeWithRank(shape_inference::InferenceContext* c, in UnchangedShapeWithRank()
183 shape_inference::InferenceContext* c, int32 rank) { in UnchangedShapeWithRankAtLeast()
191 inline Status UnchangedShapeWithRankAtMost(shape_inference::InferenceContext* c, in UnchangedShapeWithRankAtMost()
200 inline Status NoOutputs(shape_inference::InferenceContext* c) { in NoOutputs()
205 inline Status ScalarShape(shape_inference::InferenceContext* c) { in ScalarShape()
224 shape_inference::InferenceContext* context);
227 Status MatMulShape(shape_inference::InferenceContext* c);
231 Status BatchMatMulV2Shape(shape_inference::InferenceContext* c);
[all …]
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_state_ops.cc | 26 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
39 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
57 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
83 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
98 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
115 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
133 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
146 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
159 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
174 .SetShapeFn(tensorflow::shape_inference::UnknownShape)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 31 Status UnchangedRank(shape_inference::InferenceContext* c) { in UnchangedRank()
48 .SetShapeFn(shape_inference::UnknownShape)
70 .SetShapeFn(shape_inference::UnknownShape)
106 .SetShapeFn(shape_inference::UnknownShape)
169 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon00ae58f50202()
170 shape_inference::ShapeHandle lhs_shape_handle = c->input(0); in __anon00ae58f50202()
171 shape_inference::ShapeHandle rhs_shape_handle = c->input(1); in __anon00ae58f50202()
174 return shape_inference::UnknownShape(c); in __anon00ae58f50202()
200 shape_inference::DimensionOrConstant in __anon00ae58f50202()
203 shape_inference::DimensionOrConstant in __anon00ae58f50202()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 37 std::unique_ptr<shape_inference::InferenceContext> ic, const Node* node) in ExtendedInferenceContext()
52 shape_inference::InferenceContext* get_context() { in get_context()
59 std::unique_ptr<shape_inference::InferenceContext> inference_context_;
99 shape_inference::ShapeHandle shape);
112 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
161 static bool SameDefinedShape(shape_inference::InferenceContext* c,
162 shape_inference::ShapeHandle s0,
163 shape_inference::ShapeHandle s1);
168 shape_inference::InferenceContext* c,
169 const std::vector<shape_inference::ShapeAndType>& existing,
[all …]
|