| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 27 .SetShapeFn(shape_inference::ScalarShape);
32 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonf24a52a80102()
42 .SetShapeFn(shape_inference::ScalarShape);
47 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonf24a52a80202()
55 .SetShapeFn(shape_inference::UnknownShape)
63 .SetShapeFn(shape_inference::ScalarShape);
67 .SetShapeFn(shape_inference::UnknownShape);
71 .SetShapeFn(shape_inference::UnknownShape);
77 .SetShapeFn(shape_inference::UnknownShape);
83 .SetShapeFn(shape_inference::UnknownShape);
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | risc_ops.cc | 52 .SetShapeFn(shape_inference::UnchangedShape);
59 .SetShapeFn(RiscBinaryNonBroadcastOpShapeFn)
70 .SetShapeFn(shape_inference::UnchangedShape);
78 .SetShapeFn(RiscBinaryNonBroadcastOpShapeFn);
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);
[all …]
|
| D | tpu_embedding_load_retrieve_ops.cc | 40 .SetShapeFn(LoadOpShapeFunction());
52 .SetShapeFn(LoadOpShapeFunction());
63 .SetShapeFn(RetrieveOpShapeFunction());
75 .SetShapeFn(RetrieveOpShapeFunction());
85 .SetShapeFn(LoadOpShapeFunction());
96 .SetShapeFn(LoadOpShapeFunction());
106 .SetShapeFn(RetrieveOpShapeFunction());
118 .SetShapeFn(RetrieveOpShapeFunction());
130 .SetShapeFn(LoadOpShapeFunction());
143 .SetShapeFn(LoadOpShapeFunction());
[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 __anonfa0360990102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonfa0360990202()
132 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonfa0360990302()
148 .SetShapeFn(shape_inference::ScalarShape);
[all …]
|
| D | experimental_dataset_ops.cc | 26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6500dca60102()
39 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6500dca60202()
52 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6500dca60302()
68 .SetShapeFn(shape_inference::ScalarShape);
78 .SetShapeFn(shape_inference::ScalarShape);
86 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6500dca60402()
98 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon6500dca60502()
116 .SetShapeFn(shape_inference::ScalarShape);
125 .SetShapeFn(shape_inference::ScalarShape);
134 .SetShapeFn(shape_inference::ScalarShape);
[all …]
|
| D | data_flow_ops.cc | 54 .SetShapeFn([](InferenceContext* c) { in __anona2f81a440202()
148 .SetShapeFn(DynamicStitchShapeFunction);
156 .SetShapeFn(DynamicStitchShapeFunction);
191 .SetShapeFn(TwoElementOutput);
204 .SetShapeFn(shape_inference::ScalarShape);
214 .SetShapeFn(TwoElementOutput);
224 .SetShapeFn(shape_inference::ScalarShape);
234 .SetShapeFn(TwoElementOutput);
244 .SetShapeFn(shape_inference::ScalarShape);
254 .SetShapeFn(TwoElementOutput);
[all …]
|
| D | state_ops.cc | 31 .SetShapeFn(shape_inference::ExplicitShape);
40 .SetShapeFn([](InferenceContext* c) { in __anon42cd40540102()
62 .SetShapeFn(shape_inference::ScalarShape);
70 .SetShapeFn(shape_inference::ExplicitShape);
77 .SetShapeFn(shape_inference::UnchangedShape);
87 .SetShapeFn([](InferenceContext* c) { in __anon42cd40540202()
104 .SetShapeFn(shape_inference::MergeBothInputsShapeFn);
112 .SetShapeFn(shape_inference::MergeBothInputsShapeFn);
160 .SetShapeFn(ScatterUpdateShape);
170 .SetShapeFn(ScatterUpdateShape);
[all …]
|
| D | math_ops.cc | 34 .SetShapeFn([](InferenceContext* c) { in __anon57f5a5dd0102()
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);
181 .SetShapeFn(shape_inference::UnchangedShape)
194 .SetShapeFn(shape_inference::UnchangedShape);
201 .SetShapeFn(shape_inference::UnchangedShape);
[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 | io_ops.cc | 67 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0202()
93 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0302()
137 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0402()
150 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0502()
173 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0602()
199 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0702()
215 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0802()
251 .SetShapeFn(ScalarInputsAndOutputs);
257 .SetShapeFn(ScalarInputsAndOutputs);
266 .SetShapeFn(TwoElementOutput);
[all …]
|
| D | spectral_ops.cc | 31 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10102()
39 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10202()
47 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10302()
55 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10402()
63 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10502()
71 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10602()
117 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, true, 1); }); in __anon41aea5d10702()
125 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, false, 1); }); in __anon41aea5d10802()
133 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, true, 2); }); in __anon41aea5d10902()
141 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, false, 2); }); in __anon41aea5d10a02()
[all …]
|
| D | string_ops.cc | 44 .SetShapeFn([](InferenceContext* c) { in __anonf38d6eaa0102()
58 .SetShapeFn(shape_inference::UnchangedShape);
64 .SetShapeFn([](InferenceContext* c) { in __anonf38d6eaa0202()
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);
112 .SetShapeFn(shape_inference::UnsortedSegmentReductionShapeFn);
125 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
|
| D | nn_ops.cc | 80 .SetShapeFn(shape_inference::AvgPoolShape);
91 .SetShapeFn(shape_inference::AvgPoolGradShape);
106 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0202()
138 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0302()
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);
284 .SetShapeFn(shape_inference::FusedBatchNormGradShape);
[all …]
|
| D | training_ops.cc | 92 .SetShapeFn(ApplyGradientDescentShapeFn<false>);
100 .SetShapeFn(ApplyGradientDescentShapeFn<true>);
126 .SetShapeFn(ApplyProximalGradientDescentShapeFn</*is_sparse=*/false,
140 .SetShapeFn(ApplyProximalGradientDescentShapeFn</*is_sparse=*/true,
151 .SetShapeFn(ApplyProximalGradientDescentShapeFn</*is_sparse=*/false,
164 .SetShapeFn(ApplyProximalGradientDescentShapeFn</*is_sparse=*/true,
197 .SetShapeFn(
213 .SetShapeFn(
226 .SetShapeFn(
241 .SetShapeFn(ApplyAdadeltaShapeFn</*is_sparse=*/true, /*is_resource=*/true>);
[all …]
|
| D | logging_ops.cc | 31 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::UnchangedShape);
54 .SetShapeFn([](InferenceContext* c) { in __anonbfa6b29a0102()
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);
|
| D | control_flow_ops.cc | 72 .SetShapeFn(SwitchShape);
81 .SetShapeFn(SwitchShape);
89 .SetShapeFn(SwitchNShape);
98 .SetShapeFn([](InferenceContext* c) { in __anonc22003ab0202()
153 .SetShapeFn(MergeShape);
161 .SetShapeFn(MergeShape);
171 .SetShapeFn([](InferenceContext* c) { in __anonc22003ab0402()
197 .SetShapeFn(shape_inference::UnchangedShape);
204 .SetShapeFn(shape_inference::UnchangedShape);
210 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
|
| D | functional_ops.cc | 30 .SetShapeFn([](InferenceContext* c) { in __anon8af73ba60102()
63 .SetShapeFn(shape_inference::UnknownShape);
76 .SetShapeFn(shape_inference::UnknownShape)
123 .SetShapeFn(IfShapeInferenceFn);
136 .SetShapeFn(IfShapeInferenceFn);
166 .SetShapeFn(CaseShapeInferenceFn);
177 .SetShapeFn(CaseShapeInferenceFn);
187 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8af73ba60202()
244 .SetShapeFn(WhileShapeInferenceFn);
254 .SetShapeFn(WhileShapeInferenceFn);
[all …]
|
| D | resource_variable_ops.cc | 86 .SetShapeFn([](InferenceContext* c) { in __anon4d581ac00202()
108 .SetShapeFn([](InferenceContext* c) { in __anon4d581ac00302()
138 .SetShapeFn(ReadVariableShapeFn);
145 .SetShapeFn(ReadVariablesShapeFn);
167 .SetShapeFn(shape_inference::NoOutputs);
199 .SetShapeFn(CreateAssignShapeFn);
205 .SetShapeFn(CreateAssignShapeFn);
211 .SetShapeFn(CreateAssignShapeFn);
216 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
235 .SetShapeFn(VariableShapeShapeFn);
[all …]
|
| D | tpu_configuration_ops.cc | 84 .SetShapeFn([](InferenceContext* c) { in __anon6abd88830102()
115 .SetShapeFn([](InferenceContext* c) { in __anon6abd88830202()
141 .SetShapeFn([](InferenceContext* c) { in __anon6abd88830302()
156 .SetShapeFn(shape_inference::UnknownShape)
171 .SetShapeFn([](InferenceContext* c) { in __anon6abd88830402()
193 .SetShapeFn(shape_inference::UnknownShape)
210 .SetShapeFn(shape_inference::UnknownShape);
214 .SetShapeFn(shape_inference::UnknownShape);
219 .SetShapeFn(shape_inference::UnknownShape);
|
| 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);
96 .SetShapeFn(shape_inference::UnchangedShape);
105 .SetShapeFn(shape_inference::UnknownShape);
|
| D | nccl_ops.cc | 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)
130 .SetShapeFn([](InferenceContext* c) { in __anonda2b4fd30102()
|
| D | map_ops.cc | 26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon004b90f10202()
34 .SetShapeFn(shape_inference::ScalarShape);
42 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon004b90f10302()
54 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon004b90f10402()
65 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon004b90f10502()
75 .SetShapeFn(shape_inference::ScalarShape);
81 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon004b90f10602()
|
| D | tpu_infeed_ops.cc | 30 .SetShapeFn(shape_inference::ExplicitShape);
38 .SetShapeFn(shape_inference::NoOutputs)
47 .SetShapeFn(shape_inference::NoOutputs)
55 .SetShapeFn([](InferenceContext* c) { in __anon06b801f60102()
72 .SetShapeFn(shape_inference::ScalarShape);
80 .SetShapeFn(shape_inference::ScalarShape);
85 .SetShapeFn(shape_inference::NoOutputs);
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_state_ops.cc | 32 .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/ |
| D | xla_ops.cc | 50 .SetShapeFn(shape_inference::UnknownShape)
72 .SetShapeFn(shape_inference::UnknownShape)
108 .SetShapeFn(shape_inference::UnknownShape)
147 .SetShapeFn(UnchangedRank)
171 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3500048a0202()
300 .SetShapeFn(shape_inference::UnknownShape)
313 .SetShapeFn(shape_inference::UnknownShape)
326 .SetShapeFn(shape_inference::UnknownShape)
356 .SetShapeFn(shape_inference::UnchangedShape)
388 .SetShapeFn(shape_inference::UnknownShape)
[all …]
|