| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 27 .SetShapeFn(shape_inference::ScalarShape);
32 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonddbb40510102()
33 shape_inference::ShapeHandle out; in __anonddbb40510102()
42 .SetShapeFn(shape_inference::ScalarShape);
47 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonddbb40510202()
51 return shape_inference::ScalarShape(c); in __anonddbb40510202()
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 | 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 __anondee5e0a20102()
109 shape_inference::ShapeHandle count_shape; in __anondee5e0a20102()
111 return shape_inference::ScalarShape(c); in __anondee5e0a20102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20202()
[all …]
|
| D | experimental_dataset_ops.cc | 27 .SetShapeFn(shape_inference::ScalarShape);
35 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondb35634f0102()
36 shape_inference::ShapeHandle tag_shape; in __anondb35634f0102()
38 return shape_inference::ScalarShape(c); in __anondb35634f0102()
53 .SetShapeFn(shape_inference::ScalarShape);
70 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondb35634f0202()
71 shape_inference::ShapeHandle unused; in __anondb35634f0202()
86 shape_inference::ShapeHandle v; in __anondb35634f0202()
94 return shape_inference::ScalarShape(c); in __anondb35634f0202()
100 .SetShapeFn(shape_inference::ScalarShape);
[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 __anonebbd57d80202()
61 shape_inference::ShapeHandle element_shape; in __anonebbd57d80202()
65 0, std::vector<shape_inference::ShapeAndType>{ in __anonebbd57d80202()
75 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80302()
79 shape_inference::ShapeHandle element_shape = c->UnknownShape(); in __anonebbd57d80302()
[all …]
|
| D | boosted_trees_ops.cc | 26 using shape_inference::DimensionHandle;
27 using shape_inference::InferenceContext;
28 using shape_inference::ShapeHandle;
35 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0102()
36 shape_inference::ShapeHandle unused_input; in __anone3ce78ae0102()
56 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0202()
62 shape_inference::ShapeHandle node_id_range_shape; in __anone3ce78ae0202()
63 shape_inference::ShapeHandle unused_shape; in __anone3ce78ae0202()
68 shape_inference::ShapeHandle summary_shape_base; in __anone3ce78ae0202()
74 shape_inference::ShapeHandle summary_shape; in __anone3ce78ae0202()
[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);
63 .SetShapeFn(shape_inference::NoOutputs);
71 .SetShapeFn(shape_inference::NoOutputs);
79 .SetShapeFn(shape_inference::NoOutputs);
89 .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 __anonb22bfa860102()
114 .SetShapeFn(shape_inference::ExplicitShape);
174 .SetShapeFn(shape_inference::UnchangedShape);
182 .SetShapeFn(shape_inference::UnchangedShape)
195 .SetShapeFn(shape_inference::UnchangedShape);
202 .SetShapeFn(shape_inference::UnchangedShape);
211 .SetShapeFn(shape_inference::UnchangedShape)
[all …]
|
| D | logging_ops.cc | 23 using shape_inference::InferenceContext;
31 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::UnchangedShape);
76 .SetShapeFn(shape_inference::ScalarShape);
85 .SetShapeFn(shape_inference::ScalarShape);
92 .SetShapeFn(shape_inference::ScalarShape);
99 .SetShapeFn(shape_inference::ScalarShape);
111 .SetShapeFn(shape_inference::ScalarShape);
119 .SetShapeFn(shape_inference::ScalarShape);
127 .SetShapeFn(shape_inference::ScalarShape)
[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 __anon4a7ac08d0202()
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 | functional_ops.cc | 22 using shape_inference::InferenceContext;
41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anon22ceab0f0102()
63 .SetShapeFn(shape_inference::UnknownShape);
76 .SetShapeFn(shape_inference::UnknownShape)
102 .SetShapeFn(shape_inference::UnknownShape);
115 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon22ceab0f0202()
120 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in __anon22ceab0f0202()
127 shape_inference::ShapeHandle output_shape_handle; in __anon22ceab0f0202()
144 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon22ceab0f0302()
149 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in __anon22ceab0f0302()
[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 __anonc2e126fa0102()
40 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anonc2e126fa0202()
48 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anonc2e126fa0302()
56 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anonc2e126fa0402()
64 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anonc2e126fa0502()
72 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anonc2e126fa0602()
151 .SetShapeFn(shape_inference::UnknownShape)
[all …]
|
| 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 __anone3ce2a1d0102()
62 .SetShapeFn(shape_inference::ScalarShape);
70 .SetShapeFn(shape_inference::ExplicitShape);
77 .SetShapeFn(shape_inference::UnchangedShape);
91 return shape_inference::MergeBothInputsShapeFn(c); in __anone3ce2a1d0202()
104 .SetShapeFn(shape_inference::MergeBothInputsShapeFn);
112 .SetShapeFn(shape_inference::MergeBothInputsShapeFn);
[all …]
|
| D | tensor_forest_ops.cc | 30 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anoncf2fc2a90102()
31 shape_inference::ShapeHandle unused_input; in __anoncf2fc2a90102()
40 .SetShapeFn(tensorflow::shape_inference::NoOutputs);
45 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
50 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anoncf2fc2a90202()
51 shape_inference::ShapeHandle unused_input; in __anoncf2fc2a90202()
59 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
66 .SetShapeFn([](tensorflow::shape_inference::InferenceContext* c) { in __anoncf2fc2a90302()
67 shape_inference::ShapeHandle shape_handle; in __anoncf2fc2a90302()
68 shape_inference::DimensionHandle batch_size = c->UnknownDim(); in __anoncf2fc2a90302()
|
| 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 __anon130e45de0102()
58 std::vector<shape_inference::ShapeHandle> in_shapes; in __anon130e45de0102()
60 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anon130e45de0102()
69 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon130e45de0102()
70 shape_inference::DimensionOrConstant(3)})})); in __anon130e45de0102()
83 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon130e45de0202()
84 shape_inference::ShapeHandle out_shape; in __anon130e45de0202()
100 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon130e45de0302()
|
| D | nn_ops.cc | 26 using shape_inference::DimensionHandle;
27 using shape_inference::InferenceContext;
28 using shape_inference::ShapeHandle;
77 .SetShapeFn(shape_inference::AvgPoolShape);
183 .SetShapeFn(shape_inference::FusedBatchNormShape);
201 .SetShapeFn(shape_inference::FusedBatchNormShape);
218 .SetShapeFn(shape_inference::FusedBatchNormGradShape);
236 .SetShapeFn(shape_inference::FusedBatchNormGradShape);
246 .SetShapeFn(shape_inference::BiasAddShape);
254 .SetShapeFn(shape_inference::BiasAddGradShape);
[all …]
|
| /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);
230 Status BiasAddShape(shape_inference::InferenceContext* c);
[all …]
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_state_ops.cc | 26 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
42 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
68 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
83 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
100 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
118 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
131 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
144 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
159 .SetShapeFn(tensorflow::shape_inference::UnknownShape)
177 .SetShapeFn(tensorflow::shape_inference::NoOutputs)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 29 Status UnchangedRank(shape_inference::InferenceContext* c) { in UnchangedRank()
46 .SetShapeFn(shape_inference::UnknownShape)
68 .SetShapeFn(shape_inference::UnknownShape)
104 .SetShapeFn(shape_inference::UnknownShape)
167 .SetShapeFn(shape_inference::UnknownShape)
186 .SetShapeFn(shape_inference::UnknownShape)
216 .SetShapeFn(shape_inference::UnchangedShape)
248 .SetShapeFn(shape_inference::UnknownShape)
292 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc06300730202()
295 shape_inference::ShapeHandle s; in __anonc06300730202()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
| D | model_ops.cc | 29 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond0e057b60102()
30 shape_inference::ShapeHandle unused_input; in __anond0e057b60102()
43 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond0e057b60202()
44 shape_inference::ShapeHandle unused_input; in __anond0e057b60202()
61 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond0e057b60302()
62 shape_inference::ShapeHandle unused_input; in __anond0e057b60302()
78 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond0e057b60402()
79 shape_inference::ShapeHandle unused_input; in __anond0e057b60402()
97 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond0e057b60502()
98 shape_inference::ShapeHandle unused_input; in __anond0e057b60502()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 38 std::unique_ptr<shape_inference::InferenceContext> ic, const Node* node) in ExtendedInferenceContext()
57 shape_inference::InferenceContext* get_context() { in get_context()
71 std::unique_ptr<shape_inference::InferenceContext> inference_context_;
116 shape_inference::ShapeHandle shape);
129 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
191 static bool SameDefinedShape(shape_inference::InferenceContext* c,
192 shape_inference::ShapeHandle s0,
193 shape_inference::ShapeHandle s1);
198 shape_inference::InferenceContext* c,
199 const std::vector<shape_inference::ShapeAndType>& existing,
[all …]
|
| /external/tensorflow/tensorflow/contrib/bigtable/ops/ |
| D | bigtable_ops.cc | 30 .SetShapeFn(shape_inference::ScalarShape);
40 .SetShapeFn(shape_inference::ScalarShape);
48 .SetShapeFn(shape_inference::NoOutputs);
56 .SetShapeFn(shape_inference::ScalarShape);
64 .SetShapeFn(shape_inference::ScalarShape);
73 .SetShapeFn(shape_inference::ScalarShape);
80 .SetShapeFn(shape_inference::ScalarShape);
90 .SetShapeFn(shape_inference::ScalarShape);
105 .SetShapeFn(shape_inference::ScalarShape);
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
| D | array_ops.cc | 29 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon96b295990102()
32 shape_inference::ShapeHandle input_tensor_shape = c->input(0); in __anon96b295990102()
33 shape_inference::DimensionHandle num_input_elements = in __anon96b295990102()
35 shape_inference::ShapeHandle result_shape_handle; in __anon96b295990102()
36 if (!shape_inference::InferenceContext::ValueKnown(num_input_elements)) { in __anon96b295990102()
53 shape_inference::InferenceContext::Value( in __anon96b295990102()
146 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon96b295990202()
149 shape_inference::ShapeHandle s; in __anon96b295990202()
|