HomeSort by relevance Sort by last modified time
    Searched refs:TF_CALL_double (Results 1 - 25 of 51) sorted by null

1 2 3

  /external/tensorflow/tensorflow/core/kernels/
conv_ops_fused_double.cc 29 TF_CALL_double(REGISTER_FUSED_CPU_CONV2D);
38 TF_CALL_double(REGISTER_FUSED_GPU_CONV2D);
batch_matmul_op_real.cc 30 TF_CALL_double(REGISTER_BATCH_MATMUL_CPU);
39 TF_CALL_double(REGISTER_BATCH_MATMUL_GPU);
45 TF_CALL_double(REGISTER_BATCH_MATMUL_SYCL);
colorspace_op_gpu.cu.cc 31 TF_CALL_double(INSTANTIATE_GPU);
debug_ops.cc 112 TF_CALL_double(REGISTER_DEBUG_NUMERIC_SUMMARY_COUNT);
125 TF_CALL_double(REGISTER_GPU_DEBUG_NUMERIC_SUMMARY_COUNT);
139 TF_CALL_double(REGISTER_SYCL_DEBUG_NUMERIC_SUMMARY_COUNT);
reverse_op_gpu.cu.cc 46 TF_CALL_double(DEFINE_REVERSE_ALL_DIMS);
tile_ops_cpu_impl.h 37 TF_CALL_double(DEFINE_TYPE);
60 TF_CALL_double(DEFINE_TYPE);
colorspace_op.cc 120 TF_CALL_double(REGISTER_CPU);
138 TF_CALL_double(DECLARE_GPU);
148 TF_CALL_double(REGISTER_GPU);
160 TF_CALL_double(REGISTER_SYCL);
where_op.h 32 TF_CALL_double(m); \
tile_functor_cpu.cc 61 TF_CALL_double(DEFINE_TYPE);
83 TF_CALL_double(DEFINE_TYPE);
sequence_ops.cc 95 TF_CALL_double(REGISTER_SYCL_KERNEL);
102 TF_CALL_double(REGISTER_CPU_KERNEL);
109 TF_CALL_double(REGISTER_GPU_KERNEL);
173 TF_CALL_double(REGISTER_CPU_KERNEL);
179 TF_CALL_double(REGISTER_GPU_KERNEL);
185 TF_CALL_double(REGISTER_SYCL_KERNEL);
softmax_op.cc 87 TF_CALL_double(REGISTER_CPU);
96 TF_CALL_double(REGISTER_CPU);
diag_op.cc 175 TF_CALL_double(REGISTER_DIAGOP);
188 TF_CALL_double(REGISTER_DIAGPARTOP);
214 TF_CALL_double(REGISTER_DIAGOP_GPU);
237 TF_CALL_double(REGISTER_DIAGPARTOP_GPU);
tile_ops.cc 141 TF_CALL_double(HANDLE_TYPE_NAME);
219 TF_CALL_double(HANDLE_TYPE_NAME_CPU);
232 TF_CALL_double(HANDLE_TYPE_NAME_GPU);
243 TF_CALL_double(HANDLE_TYPE_NAME_SYCL);
325 TF_CALL_double(HANDLE_TYPE_NAME);
487 TF_CALL_double(HANDLE_TYPE_NAME_CPU);
497 TF_CALL_double(HANDLE_TYPE_NAME_GPU);
511 TF_CALL_double(HANDLE_TYPE_NAME_SYCL);
579 TF_CALL_double(REGISTER_GPU);
620 TF_CALL_double(REGISTER_SYCL)
    [all...]
batch_norm_op.cc 175 TF_CALL_double(REGISTER_KERNEL);
219 TF_CALL_double(REGISTER_KERNEL);
231 TF_CALL_double(REGISTER_KERNEL);
278 TF_CALL_double(REGISTER_KERNEL);
conditional_accumulator_op.cc 57 TF_CALL_double(REGISTER_KERNELS_CPU);
sparse_conditional_accumulator_op.cc 57 TF_CALL_double(REGISTER_KERNELS_CPU);
multinomial_op.cc 253 TF_CALL_double(REGISTER);
273 TF_CALL_double(REGISTER);
321 TF_CALL_double(REGISTER);
343 TF_CALL_double(REGISTER);
stateful_random_ops.cc 214 TF_CALL_double(REGISTER_CPU);
220 TF_CALL_double(REGISTER_GPU);
stateless_random_ops.cc 199 TF_CALL_double(REGISTER_CPU);
207 TF_CALL_double(REGISTER_GPU);
depthwise_conv_op.cc 461 TF_CALL_double(REGISTER_CPU_KERNEL);
473 TF_CALL_double(REGISTER_GPU_KERNEL);
495 TF_CALL_double(REGISTER_GROUPED_CONV_KERNEL);
ragged_range_op.cc 122 TF_CALL_double(REGISTER_CPU_KERNEL);
  /external/tensorflow/tensorflow/core/framework/
register_types.h 26 // - TF_CALL_float, TF_CALL_double, etc. which call the given macro with
30 // into TF_CALL_float, TF_CALL_double, etc. so they filter by target platform
63 #define TF_CALL_double(m) m(double)
94 #define TF_CALL_double(m)
124 #define TF_CALL_double(m)
165 TF_CALL_half(m) TF_CALL_bfloat16(m) TF_CALL_float(m) TF_CALL_double(m)
172 TF_CALL_INTEGRAL_TYPES(m) TF_CALL_half(m) TF_CALL_float(m) TF_CALL_double(m)
175 TF_CALL_half(m) TF_CALL_bfloat16(m) TF_CALL_float(m) TF_CALL_double(m) \
198 TF_CALL_half(m) TF_CALL_float(m) TF_CALL_double(m)
205 #define TF_CALL_GPU_NUMBER_TYPES_NO_HALF(m) TF_CALL_float(m) TF_CALL_double(m
    [all...]
register_types_traits.h 102 TF_CALL_double(m) TF_CALL_float(m) TF_CALL_half(m) TF_CALL_int32(m) \
106 TF_CALL_double(m) TF_CALL_float(m) TF_CALL_int32(m)
  /external/tensorflow/tensorflow/contrib/image/kernels/
image_ops.cc 137 TF_CALL_double(REGISTER);
161 TF_CALL_double(DECLARE_FUNCTOR);
181 TF_CALL_double(REGISTER);
  /external/tensorflow/tensorflow/core/kernels/data/experimental/sql/
sqlite_query_connection.cc 102 TF_CALL_double(DOUBLE_CASE)

Completed in 3674 milliseconds

1 2 3