Home
last modified time | relevance | path

Searched refs:ScalarShape (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
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);
111 return shape_inference::ScalarShape(c); in __anondee5e0a20102()
123 return shape_inference::ScalarShape(c); in __anondee5e0a20202()
135 return shape_inference::ScalarShape(c); in __anondee5e0a20302()
148 .SetShapeFn(shape_inference::ScalarShape);
[all …]
Dexperimental_dataset_ops.cc27 .SetShapeFn(shape_inference::ScalarShape);
38 return shape_inference::ScalarShape(c); in __anondb35634f0102()
53 .SetShapeFn(shape_inference::ScalarShape);
94 return shape_inference::ScalarShape(c); in __anondb35634f0202()
100 .SetShapeFn(shape_inference::ScalarShape);
126 return shape_inference::ScalarShape(c); in __anondb35634f0302()
136 .SetShapeFn(shape_inference::ScalarShape);
156 .SetShapeFn(shape_inference::ScalarShape);
173 .SetShapeFn(shape_inference::ScalarShape);
180 .SetShapeFn(shape_inference::ScalarShape);
[all …]
Dlogging_ops.cc76 .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)
134 .SetShapeFn(shape_inference::ScalarShape);
139 .SetShapeFn(shape_inference::ScalarShape);
Ddata_flow_ops.cc204 .SetShapeFn(shape_inference::ScalarShape);
224 .SetShapeFn(shape_inference::ScalarShape);
244 .SetShapeFn(shape_inference::ScalarShape);
264 .SetShapeFn(shape_inference::ScalarShape);
383 .SetShapeFn(shape_inference::ScalarShape);
388 .SetShapeFn(shape_inference::ScalarShape);
405 .SetShapeFn(shape_inference::ScalarShape);
682 return shape_inference::ScalarShape(c); in __anon4a7ac08d1402()
771 return shape_inference::ScalarShape(c); in __anon4a7ac08d1702()
808 return shape_inference::ScalarShape(c); in __anon4a7ac08d1902()
[all …]
Dtpu_infeed_ops.cc72 .SetShapeFn(shape_inference::ScalarShape);
80 .SetShapeFn(shape_inference::ScalarShape);
Dtensor_forest_ops.cc45 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
59 .SetShapeFn(tensorflow::shape_inference::ScalarShape);
Dio_ops.cc273 .SetShapeFn(shape_inference::ScalarShape);
290 .SetShapeFn(shape_inference::ScalarShape);
314 .SetShapeFn(shape_inference::ScalarShape);
331 .SetShapeFn(shape_inference::ScalarShape);
353 .SetShapeFn(shape_inference::ScalarShape);
Dtpu_heartbeat_ops.cc26 .SetShapeFn(shape_inference::ScalarShape);
Dclustering_ops.cc33 .SetShapeFn(shape_inference::ScalarShape);
/external/tensorflow/tensorflow/contrib/bigtable/ops/
Dbigtable_ops.cc30 .SetShapeFn(shape_inference::ScalarShape);
40 .SetShapeFn(shape_inference::ScalarShape);
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);
Dbigtable_test_ops.cc25 .SetShapeFn(shape_inference::ScalarShape);
/external/tensorflow/tensorflow/compiler/xrt/ops/
Dxrt_state_ops.cc26 .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)
Dxrt_execute_op.cc37 return tensorflow::shape_inference::ScalarShape(c); in __anon8fb216580102()
/external/tensorflow/tensorflow/contrib/memory_stats/ops/
Dmemory_stats_ops.cc23 .SetShapeFn(shape_inference::ScalarShape);
27 .SetShapeFn(shape_inference::ScalarShape);
31 .SetShapeFn(shape_inference::ScalarShape);
/external/tensorflow/tensorflow/contrib/tensor_forest/ops/
Dmodel_ops.cc31 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
52 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
76 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
Dstats_ops.cc30 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
52 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
/external/tensorflow/tensorflow/python/framework/
Dtest_ops.cc27 .SetShapeFn(shape_inference::ScalarShape);
42 .SetShapeFn(shape_inference::ScalarShape);
51 return shape_inference::ScalarShape(c); in __anonddbb40510202()
63 .SetShapeFn(shape_inference::ScalarShape);
669 .SetShapeFn(shape_inference::ScalarShape);
/external/tensorflow/tensorflow/contrib/hadoop/ops/
Ddataset_ops.cc27 .SetShapeFn(shape_inference::ScalarShape);
/external/tensorflow/tensorflow/contrib/kinesis/ops/
Ddataset_ops.cc29 .SetShapeFn(shape_inference::ScalarShape)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/ops/
Dget_serialized_resource_op.cc30 .SetShapeFn(shape_inference::ScalarShape)
/external/tensorflow/tensorflow/contrib/kafka/ops/
Ddataset_ops.cc30 .SetShapeFn(shape_inference::ScalarShape)
/external/tensorflow/tensorflow/contrib/ffmpeg/
Dencode_audio_op.cc128 .SetShapeFn(shape_inference::ScalarShape)
206 .SetShapeFn(shape_inference::ScalarShape)
/external/tensorflow/tensorflow/contrib/ignite/ops/
Ddataset_ops.cc33 .SetShapeFn(shape_inference::ScalarShape)
/external/tensorflow/tensorflow/contrib/coder/ops/
Dcoder_ops.cc32 .SetShapeFn(shape_inference::ScalarShape)
/external/tensorflow/tensorflow/go/
Dshape.go34 func ScalarShape() Shape { func

12