| /external/tensorflow/tensorflow/core/kernels/ |
| quantized_conv_ops_test.cc | 51 .Attr("out_type", DataTypeToEnum<qint32>::v())
143 .Attr("out_type", DataTypeToEnum<qint32>::v())
186 .Attr("out_type", DataTypeToEnum<qint32>::v())
227 .Attr("out_type", DataTypeToEnum<qint32>::v())
272 .Attr("out_type", DataTypeToEnum<qint32>::v())
|
| pooling_ops_common.cc | 165 DataTypeToEnum<T>::value,
178 DataTypeToEnum<T>::value,
270 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value,
281 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::value,
290 context->allocate_temp(DataTypeToEnum<T>::value,
299 context->allocate_temp(DataTypeToEnum<T>::value,
|
| quantized_bias_add_op_test.cc | 48 .Attr("out_type", DataTypeToEnum<qint32>::v())
98 .Attr("out_type", DataTypeToEnum<qint32>::v())
|
| scatter_nd_op.cc | 55 const DataType dt = DataTypeToEnum<T>::v();
56 const DataType index_t = DataTypeToEnum<Index>::v();
86 const DataType dt = DataTypeToEnum<T>::v();
87 const DataType dt_ref = DataTypeToEnum<T>::ref();
88 const DataType index_t = DataTypeToEnum<Index>::v();
361 DataTypeString(DataTypeToEnum<Index>::v()),
367 DataTypeString(DataTypeToEnum<Index>::v()),
451 TF_RETURN_IF_ERROR(c->allocate_temp(DataTypeToEnum<T>::value, shape, out));
|
| self_adjoint_eig_v2_op_gpu.cc | 93 solver->allocate_scoped_tensor(DataTypeToEnum<RealScalar>::value,
102 {0}, DataTypeToEnum<Scalar>::value, input.shape(), &input_copy),
|
| unravel_index_op.cc | 61 ctx->allocate_temp(DataTypeToEnum<Tidx>::value,
72 ctx->allocate_temp(DataTypeToEnum<Tidx>::value,
|
| bincount_op_gpu.cu.cc | 79 DataTypeToEnum<int8>::value,
|
| inplace_ops_functor_gpu.cu.cc | 64 case DataTypeToEnum<type>::value: \
|
| matrix_solve_op.cc | 182 solver->allocate_scoped_tensor(DataTypeToEnum<Scalar>::value,
191 {0}, DataTypeToEnum<Scalar>::value, input.shape(), &input_copy),
206 solver->allocate_scoped_tensor(DataTypeToEnum<int>::value,
255 solver->allocate_scoped_tensor(DataTypeToEnum<Scalar>::value,
|
| requantization_range_op_test.cc | 47 .Attr("Tinput", DataTypeToEnum<qint32>::v())
|
| resize_bicubic_op_test.cc | 43 DataTypeToEnum<float>::v(), shape);
48 DataTypeToEnum<float>::v());
51 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v());
172 DataTypeToEnum<float>::v(),
|
| scatter_op.cc | 108 DataTypeString(DataTypeToEnum<Index>::v()),
115 DataTypeString(DataTypeToEnum<Index>::v()),
172 DataTypeString(DataTypeToEnum<Index>::v()),
179 DataTypeString(DataTypeToEnum<Index>::v()),
|
| ops_testutil.h | 117 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping);
124 test::FillValues<T>(AddInput(DataTypeToEnum<T>::v(), shape), data);
132 test::FillValues<T>(AddInput(DataTypeToEnum<T>::v(), shape), data);
|
| reduction_ops_common.h | 141 const DataType dt = DataTypeToEnum<T>::v();
142 const DataType pt = DataTypeToEnum<Tperm>::v();
222 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value,
|
| sparse_add_op_test.cc | 34 DataType value_type = tensorflow::DataTypeToEnum<T>::value;
101 DataType val_dtype = tensorflow::DataTypeToEnum<VALTYPE>::value; \
159 DataType val_dtype = tensorflow::DataTypeToEnum<VALTYPE>::value; \
|
| resize_bilinear_op_test.cc | 49 DataTypeToEnum<float>::v(), shape);
54 DataTypeToEnum<float>::v());
57 CHECK_EQ(input_types_[inputs_.size()], DataTypeToEnum<float>::v());
120 DataTypeToEnum<float>::v(),
192 DataTypeToEnum<float>::v(), TensorShape({1, 1, 1, 1})));
|
| conv_grad_ops_3d.cc | 194 context->allocate_temp(DataTypeToEnum<T>::v(),
209 OP_REQUIRES_OK(context, context->allocate_temp(DataTypeToEnum<T>::v(),
317 context->allocate_temp(DataTypeToEnum<T>::v(),
336 context->allocate_temp(DataTypeToEnum<T>::v(),
356 context, context->allocate_temp(DataTypeToEnum<T>::v(),
594 context->allocate_temp(DataTypeToEnum<T>::value,
609 DataTypeToEnum<T>::value, nchw_shape,
624 context->allocate_temp(DataTypeToEnum<T>::value, compatible_input_shape,
[all...] |
| list_kernels.h | 162 if (tmp.IsAligned() || !DataTypeCanUseMemcpy(DataTypeToEnum<T>::value)) {
227 case DataTypeToEnum<dtype>::value: \
254 case DataTypeToEnum<dtype>::value: \
|
| concat_lib_cpu.cc | 29 if (DataTypeCanUseMemcpy(DataTypeToEnum<T>::v())) {
|
| conditional_accumulator.h | 104 Tensor c(DataTypeToEnum<T>::value, {});
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops.cc | 132 DataTypeToEnum<T>::v(),
138 DataTypeToEnum<T>::v(),
144 ctx->allocate_temp(DataTypeToEnum<T>::v(),
330 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(),
335 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(),
340 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::v(),
346 DataTypeToEnum<T>::v(),
352 DataTypeToEnum<T>::v(),
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_slice_writer.h | 107 DataType dt = DataTypeToEnum<T>::value;
168 (MaxBytesPerElement(DataTypeToEnum<T>::value) * num_elements);
|
| /external/tensorflow/tensorflow/cc/framework/ |
| ops.h | 113 Tensor t(DataTypeToEnum<RealT>::v(), TensorShape());
126 Tensor t(DataTypeToEnum<RealT>::v(), shape);
140 Tensor t(DataTypeToEnum<RealT>::v(),
152 Tensor t(DataTypeToEnum<RealT>::v(), shape);
|
| /external/tensorflow/tensorflow/contrib/util/ |
| convert_graphdef_memmapped_format_lib.cc | 58 #define TYPE_FOR_SET(type) static_cast<int>(DataTypeToEnum<type>::value),
|
| /external/tensorflow/tensorflow/core/kernels/data/sql/ |
| sqlite_query_connection.cc | 86 case DataTypeToEnum<T>::value: \
|
