| /external/tensorflow/tensorflow/cc/ops/ |
| const_op.h | 42 if (val.tensor.dtype() == DataTypeToEnum<DstT>::v()) {
46 Tensor t(DataTypeToEnum<DstT>::v(), val.tensor.shape());
57 .Attr("DstT", DataTypeToEnum<DstT>::v());
|
| /external/tensorflow/tensorflow/core/framework/ |
| kernel_def_builder.h | 82 return this->TypeConstraint(attr_name, DataTypeToEnum<T>::v());
|
| numeric_op.h | 33 const DataType dt = DataTypeToEnum<T>::v();
43 const DataType dt = DataTypeToEnum<T>::v();
|
| tensor_testutil.h | 32 Tensor ret(DataTypeToEnum<T>::value, {});
40 Tensor ret(DataTypeToEnum<T>::value, {static_cast<int64>(vals.size())});
167 ASSERT_EQ(x.dtype(), DataTypeToEnum<T>::v());
184 ASSERT_EQ(x.dtype(), DataTypeToEnum<T>::v());
202 ASSERT_EQ(x.dtype(), DataTypeToEnum<T>::v());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| quantize_down_and_shrink_range_op_test.cc | 43 .Attr("Tinput", DataTypeToEnum<qint32>::v())
44 .Attr("out_type", DataTypeToEnum<quint8>::v())
|
| quantize_op_test.cc | 36 .Attr("T", DataTypeToEnum<quint8>::v())
59 .Attr("T", DataTypeToEnum<quint8>::v())
91 .Attr("T", DataTypeToEnum<quint8>::v())
121 .Attr("T", DataTypeToEnum<qint8>::v())
151 .Attr("T", DataTypeToEnum<qint8>::v())
179 .Attr("T", DataTypeToEnum<qint8>::v())
209 .Attr("T", DataTypeToEnum<qint8>::v())
239 .Attr("T", DataTypeToEnum<qint32>::v())
280 .Attr("T", DataTypeToEnum<quint8>::v())
303 .Attr("T", DataTypeToEnum<quint8>::v()
[all...] |
| requantize_op_test.cc | 40 .Attr("Tinput", DataTypeToEnum<qint32>::v())
41 .Attr("out_type", DataTypeToEnum<quint8>::v())
|
| quantize_and_dequantize_op.cc | 71 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value,
73 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value,
130 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value,
132 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value,
174 Tensor input_min_tensor(DataTypeToEnum<T>::value, TensorShape());
175 Tensor input_max_tensor(DataTypeToEnum<T>::value, TensorShape());
|
| reverse_op_test.cc | 52 MakeOp(DataTypeToEnum<T>::value);
58 Tensor expected(allocator(), DataTypeToEnum<T>::value, TensorShape({}));
65 MakeOp(DataTypeToEnum<T>::value);
78 Tensor expected(allocator(), DataTypeToEnum<T>::value,
91 MakeOp(DataTypeToEnum<T>::value);
104 Tensor expected(allocator(), DataTypeToEnum<T>::value,
188 Tensor data(DataTypeToEnum<T>::value, shape);
|
| quantized_matmul_op_test.cc | 52 .Attr("Toutput", DataTypeToEnum<qint32>::v())
103 .Attr("Toutput", DataTypeToEnum<qint32>::v())
153 .Attr("Toutput", DataTypeToEnum<qint32>::v())
193 .Attr("Toutput", DataTypeToEnum<qint32>::v())
282 .Attr("Toutput", DataTypeToEnum<qint32>::v())
|
| cudnn_pooling_gpu.cc | 49 DataTypeToEnum<T>::value,
63 DataTypeToEnum<T>::value,
138 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value,
149 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value,
158 context->allocate_temp(DataTypeToEnum<T>::value,
167 context->allocate_temp(DataTypeToEnum<T>::value,
|
| histogram_op_gpu.cu.cc | 50 DataTypeToEnum<T>::value, TensorShape({nbins + 1}), &levels_tensor,
90 DataTypeToEnum<int8>::value,
|
| listdiff_op.cc | 31 const DataType dt = DataTypeToEnum<T>::v();
32 const DataType dtidx = DataTypeToEnum<Tidx>::v();
|
| quantized_pooling_ops_test.cc | 45 .Attr("T", DataTypeToEnum<quint8>::v())
90 .Attr("T", DataTypeToEnum<quint8>::v())
|
| sparse_reduce_sum_op_test.cc | 34 DataType value_type = tensorflow::DataTypeToEnum<T>::value;
75 DataType value_type = tensorflow::DataTypeToEnum<T>::value;
|
| tile_ops.cc | 134 #define HANDLE_TYPE_NAME(T) HANDLE_TYPE(DataTypeToEnum<T>::value)
199 HANDLE_CASE(CPUDevice, DataTypeToEnum<T>::value, int32); \
200 HANDLE_CASE(CPUDevice, DataTypeToEnum<T>::value, int64);
203 HANDLE_CASE(GPUDevice, DataTypeToEnum<T>::value, int32); \
204 HANDLE_CASE(GPUDevice, DataTypeToEnum<T>::value, int64);
208 HANDLE_CASE(SYCLDevice, DataTypeToEnum<T>::value, int32); \
209 HANDLE_CASE(SYCLDevice, DataTypeToEnum<T>::value, int64);
317 #define HANDLE_TYPE_NAME(T) HANDLE_TYPE(DataTypeToEnum<T>::value)
474 HANDLE_CASE_DIM(CPUDevice, T, DataTypeToEnum<T>::value);
477 HANDLE_CASE_DIM(GPUDevice, T, DataTypeToEnum<T>::value)
[all...] |
| batch_util.cc | 99 case DataTypeToEnum<T>::value: { \
119 case DataTypeToEnum<T>::value: { \
174 case DataTypeToEnum<T>::value: { \
219 if (element->dtype() == DataTypeToEnum<T>::value) { \
|
| cast_op_test.cc | 33 Tensor data(DataTypeToEnum<Src>::value,
37 DataTypeToEnum<Dst>::value);
54 DataType in_type = DataTypeToEnum<INPUT>::v();
55 DataType out_type = DataTypeToEnum<OUTPUT>::v();
|
| dequantize_op_test.cc | 65 .Input(FakeInput(DataTypeToEnum<T>::v()))
68 .Attr("T", DataTypeToEnum<T>::v())
93 .Input(FakeInput(DataTypeToEnum<T>::v()))
96 .Attr("T", DataTypeToEnum<T>::v())
|
| gather_nd_op.cc | 39 const DataType dt = DataTypeToEnum<T>::v();
40 const DataType index_t = DataTypeToEnum<Index>::v();
111 DataTypeString(DataTypeToEnum<Index>::v()),
144 c->allocate_temp(DataTypeToEnum<T>::value, result_shape, out));
|
| lookup_table_op.h | 82 *table, DataTypeToEnum<key_dtype>::v(),
83 DataTypeToEnum<value_dtype>::v(), cinfo_.name()));
178 DataType key_dtype() const override { return DataTypeToEnum<K>::v(); }
180 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); }
|
| resize_area_op_test.cc | 43 DataTypeToEnum<float>::v(), shape);
48 DataTypeToEnum<float>::v());
51 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v());
154 DataTypeToEnum<float>::v(),
|
| slice_op_test.cc | 42 DataType dt = DataTypeToEnum<T>::v();
|
| substr_op.cc | 126 context->allocate_temp(DataTypeToEnum<T>::v(),
136 context->allocate_temp(DataTypeToEnum<T>::v(),
175 context->allocate_temp(DataTypeToEnum<T>::v(),
185 context->allocate_temp(DataTypeToEnum<T>::v(),
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| periodic_resample_op.h | 200 case tensorflow::DataTypeToEnum<float>::value:
203 case tensorflow::DataTypeToEnum<double>::value:
206 case tensorflow::DataTypeToEnum<tensorflow::int32>::value:
210 case tensorflow::DataTypeToEnum<tensorflow::int64>::value:
|
