| /external/tensorflow/tensorflow/core/kernels/ |
| batch_norm_op_test.cc | 64 .Input(FakeInput(DT_HALF))
65 .Input(FakeInput(DT_HALF))
66 .Input(FakeInput(DT_HALF))
67 .Input(FakeInput(DT_HALF))
68 .Input(FakeInput(DT_HALF))
81 Tensor expected(allocator(), DT_HALF, TensorShape({1, 1, 6, 2}));
|
| spacetobatch_benchmark_test.cc | 45 } else if (dtype == DT_HALF) {
46 Tensor input(DT_HALF, input_shape);
76 BM_Expand(BM_SpaceToBatchDev(OP, cpu, DT_HALF, __VA_ARGS__)); \
77 BM_Expand(BM_SpaceToBatchDev(OP, gpu, DT_HALF, __VA_ARGS__));
|
| cwise_ops_test.cc | 167 BM_BIAS_ADD_ALL(cpu, half, DT_HALF);
169 BM_BIAS_ADD_ALL(gpu, half, DT_HALF);
219 BM_BIAS_ADD_GRAD_ALL(gpu, NCHW, half, DT_HALF);
225 BM_BIAS_ADD_GRAD_ALL(cpu, NHWC, half, DT_HALF);
227 BM_BIAS_ADD_GRAD_ALL(gpu, NHWC, half, DT_HALF);
|
| cast_op.cc | 99 } else if (src_dtype_ == DT_HALF) {
148 } else if (src_dtype_ == DT_HALF) {
|
| restore_v2_op_test.cc | 80 DT_INT16, DT_COMPLEX64, DT_HALF})) // tensors
89 DT_INT16, DT_COMPLEX64, DT_HALF})) // tensors
336 MakeRestoreOp(DT_HALF);
|
| save_v2_op_test.cc | 46 DT_COMPLEX128, DT_HALF})) // tensors
334 EXPECT_EQ(DT_HALF, val.dtype());
|
| summary_op_test.cc | 99 MakeOp(DT_HALF);
224 MakeOp(DT_HALF);
|
| transpose_functor.h | 183 case DT_HALF:
|
| restore_op_test.cc | 79 DT_COMPLEX64, DT_HALF}))
337 MakeRestoreOp(DT_HALF);
|
| save_op_test.cc | 52 DT_STRING, DT_COMPLEX64, DT_COMPLEX128, DT_HALF}))
381 EXPECT_EQ(DT_HALF, type);
|
| summary_image_op.cc | 91 } else if (tensor.dtype() == DT_HALF) {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| type_util.cc | 55 case tensorflow::DT_HALF:
|
| xla_op_registry.h | 49 {DT_HALF, DT_FLOAT, DT_DOUBLE, DT_BFLOAT16}};
51 {DT_UINT32, DT_UINT64, DT_INT32, DT_INT64, DT_HALF, DT_FLOAT, DT_DOUBLE,
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| sycl_util.h | 60 case DT_HALF:
|
| sycl_device_context.cc | 54 case DT_HALF:
132 case DT_HALF:
|
| /external/tensorflow/tensorflow/core/framework/ |
| types.h | 264 ToSet(DT_QUINT16) | ToSet(DT_QINT32) | ToSet(DT_HALF) | ToSet(DT_RESOURCE) |
275 ToSet(DT_HALF) | ToSet(DT_UINT32) | ToSet(DT_UINT64) | ToSet(DT_BFLOAT16);
285 ToSet(DT_QUINT8) | ToSet(DT_QINT32) | ToSet(DT_HALF) | ToSet(DT_UINT32) |
299 ToSet(DT_QINT32) | ToSet(DT_HALF) | ToSet(DT_BFLOAT16);
307 ToSet(DT_FLOAT) | ToSet(DT_INT32) | ToSet(DT_INT64) | ToSet(DT_HALF);
312 ToSet(DT_QUINT8) | ToSet(DT_QINT32) | ToSet(DT_HALF);
323 ToSet(DT_HALF);
401 MATCH_TYPE_AND_ENUM(Eigen::half, DT_HALF);
425 ToSet(DT_BFLOAT16) | ToSet(DT_HALF);
432 ToSet(DT_HALF) | ToSet(DT_BFLOAT16) | ToSet(DT_FLOAT) | ToSet(DT_DOUBLE)
[all...] |
| types.cc | 95 case DT_HALF:
189 *dt = DT_HALF;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| matmul_op.cc | 27 {DT_HALF, DT_BFLOAT16, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64}};
|
| scan_ops.cc | 41 {DT_HALF, DT_BFLOAT16, DT_FLOAT}};
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_tensor_coding_test.cc | 98 DoTest<Eigen::half>(DT_HALF);
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_slice_writer.cc | 167 case DT_HALF:
|
| /external/tensorflow/tensorflow/python/framework/ |
| dtypes.py | 329 float16 = DType(types_pb2.DT_HALF)
402 types_pb2.DT_HALF: float16,
452 types_pb2.DT_HALF: "float16",
553 types_pb2.DT_HALF:
|
| /external/tensorflow/tensorflow/python/framework/testdata/ |
| metrics_export_meta_graph.pb | 73 type: DT_HALF
190 type: DT_HALF
282 type: DT_HALF
359 type: DT_HALF
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| tensor_coding_test.cc | 154 DoTest<Eigen::half>(DT_HALF);
|
| /external/tensorflow/tensorflow/contrib/summary/ |
| summary_ops_test.py | 42 types_pb2.DT_HALF: np.float16,
|
