/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:
|