Home
last modified time | relevance | path

Searched refs:UnchangedShape (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Drisc_ops.cc52 .SetShapeFn(shape_inference::UnchangedShape);
70 .SetShapeFn(shape_inference::UnchangedShape);
102 .SetShapeFn(shape_inference::UnchangedShape);
108 .SetShapeFn(shape_inference::UnchangedShape);
153 .SetShapeFn(shape_inference::UnchangedShape);
175 .SetShapeFn(shape_inference::UnchangedShape);
188 .SetShapeFn(shape_inference::UnchangedShape);
207 .SetShapeFn(shape_inference::UnchangedShape);
213 .SetShapeFn(shape_inference::UnchangedShape);
219 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
Dstring_ops.cc58 .SetShapeFn(shape_inference::UnchangedShape);
75 .SetShapeFn(shape_inference::UnchangedShape);
81 .SetShapeFn(shape_inference::UnchangedShape);
88 .SetShapeFn(shape_inference::UnchangedShape);
94 .SetShapeFn(shape_inference::UnchangedShape);
125 .SetShapeFn(shape_inference::UnchangedShape);
223 .SetShapeFn(shape_inference::UnchangedShape);
229 .SetShapeFn(shape_inference::UnchangedShape);
234 .SetShapeFn(shape_inference::UnchangedShape);
240 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
Ddebug_ops.cc35 .SetShapeFn(shape_inference::UnchangedShape);
44 .SetShapeFn(shape_inference::UnchangedShape);
55 .SetShapeFn(shape_inference::UnchangedShape);
96 .SetShapeFn(shape_inference::UnchangedShape);
Dnccl_ops.cc32 .SetShapeFn(shape_inference::UnchangedShape);
43 .SetShapeFn(shape_inference::UnchangedShape);
75 .SetShapeFn(shape_inference::UnchangedShape)
100 .SetShapeFn(shape_inference::UnchangedShape);
Dmath_ops.cc173 .SetShapeFn(shape_inference::UnchangedShape);
181 .SetShapeFn(shape_inference::UnchangedShape)
194 .SetShapeFn(shape_inference::UnchangedShape);
201 .SetShapeFn(shape_inference::UnchangedShape);
210 .SetShapeFn(shape_inference::UnchangedShape)
216 .SetShapeFn(shape_inference::UnchangedShape)
222 .SetShapeFn(shape_inference::UnchangedShape)
229 .SetShapeFn(shape_inference::UnchangedShape)
305 .SetShapeFn(shape_inference::UnchangedShape)
319 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
Dcontrol_flow_ops.cc197 .SetShapeFn(shape_inference::UnchangedShape);
204 .SetShapeFn(shape_inference::UnchangedShape);
210 .SetShapeFn(shape_inference::UnchangedShape);
217 .SetShapeFn(shape_inference::UnchangedShape);
223 .SetShapeFn(shape_inference::UnchangedShape);
Dbitwise_ops.cc27 .SetShapeFn(shape_inference::UnchangedShape);
48 .SetShapeFn(shape_inference::UnchangedShape);
Dcollective_ops.cc36 .SetShapeFn(shape_inference::UnchangedShape);
121 .SetShapeFn(shape_inference::UnchangedShape);
158 .SetShapeFn(shape_inference::UnchangedShape);
Dmkl_array_ops.cc34 using shape_inference::UnchangedShape;
119 TF_RETURN_IF_ERROR(shape_inference::UnchangedShape(c)); in __anon741deadc0302()
150 TF_RETURN_IF_ERROR(shape_inference::UnchangedShape(c)); in __anon741deadc0402()
Dtpu_cross_replica_ops.cc82 .SetShapeFn(shape_inference::UnchangedShape);
89 .SetShapeFn(shape_inference::UnchangedShape);
Darray_ops.cc37 using shape_inference::UnchangedShape;
374 .SetShapeFn(UnchangedShape);
382 .SetShapeFn(UnchangedShape);
390 .SetShapeFn(UnchangedShape);
398 .SetShapeFn(UnchangedShape);
744 return UnchangedShape(c); in __anon42d741190d02()
754 .SetShapeFn(shape_inference::UnchangedShape);
763 .SetShapeFn(shape_inference::UnchangedShape);
960 .SetShapeFn(shape_inference::UnchangedShape);
1138 .SetShapeFn(shape_inference::UnchangedShape)
[all …]
Dnn_ops.cc524 .SetShapeFn(shape_inference::UnchangedShape);
532 .SetShapeFn(shape_inference::UnchangedShape);
1064 .SetShapeFn(shape_inference::UnchangedShape);
1086 .SetShapeFn(shape_inference::UnchangedShape);
1099 .SetShapeFn(shape_inference::UnchangedShape);
1113 .SetShapeFn(shape_inference::UnchangedShape);
1127 .SetShapeFn(shape_inference::UnchangedShape);
1140 .SetShapeFn(shape_inference::UnchangedShape);
1153 .SetShapeFn(shape_inference::UnchangedShape);
1166 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
Dspecial_math_ops.cc27 .SetShapeFn(shape_inference::UnchangedShape)
Dmanip_ops.cc41 return shape_inference::UnchangedShape(c); in __anonf8805fef0102()
Dlogging_ops.cc45 .SetShapeFn(shape_inference::UnchangedShape);
Dparsing_ops.cc143 .SetShapeFn(shape_inference::UnchangedShape);
468 .SetShapeFn(shape_inference::UnchangedShape);
500 .SetShapeFn(shape_inference::UnchangedShape);
Drandom_ops.cc112 .SetShapeFn(shape_inference::UnchangedShape);
Dstate_ops.cc77 .SetShapeFn(shape_inference::UnchangedShape);
Dimage_ops.cc678 return shape_inference::UnchangedShape(c); in __anon30b5031d0f02()
Ddata_flow_ops.cc398 .SetShapeFn(shape_inference::UnchangedShape);
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx_test.cc51 .SetShapeFn(shape_inference::UnchangedShape);
60 .SetShapeFn(shape_inference::UnchangedShape);
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc356 .SetShapeFn(shape_inference::UnchangedShape)
697 .SetShapeFn(shape_inference::UnchangedShape)
854 return shape_inference::UnchangedShape(c); in __anonbbc1c2d10c02()
903 .SetShapeFn(shape_inference::UnchangedShape)
946 .SetShapeFn(shape_inference::UnchangedShape)
/external/tensorflow/tensorflow/examples/adding_an_op/
Dzero_out_op_kernel_2.cc28 .SetShapeFn(shape_inference::UnchangedShape)
/external/tensorflow/tensorflow/core/common_runtime/
Dtestlib_ops.cc85 .SetShapeFn(shape_inference::UnchangedShape);
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h47 Status UnchangedShape(shape_inference::InferenceContext* c);

12