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

1 2 3

  /external/tensorflow/tensorflow/core/kernels/
conv_ops_fused_float.cc 29 TF_CALL_float(REGISTER_FUSED_CPU_CONV2D);
38 TF_CALL_float(REGISTER_FUSED_GPU_CONV2D);
batch_matmul_op_real.cc 29 TF_CALL_float(REGISTER_BATCH_MATMUL_CPU);
38 TF_CALL_float(REGISTER_BATCH_MATMUL_GPU);
44 TF_CALL_float(REGISTER_BATCH_MATMUL_SYCL);
colorspace_op_gpu.cu.cc 30 TF_CALL_float(INSTANTIATE_GPU);
debug_ops.cc 111 TF_CALL_float(REGISTER_DEBUG_NUMERIC_SUMMARY_COUNT);
124 TF_CALL_float(REGISTER_GPU_DEBUG_NUMERIC_SUMMARY_COUNT);
138 TF_CALL_float(REGISTER_SYCL_DEBUG_NUMERIC_SUMMARY_COUNT);
reverse_op_gpu.cu.cc 45 TF_CALL_float(DEFINE_REVERSE_ALL_DIMS);
tile_ops_cpu_impl.h 35 TF_CALL_float(DEFINE_TYPE);
58 TF_CALL_float(DEFINE_TYPE);
colorspace_op.cc 119 TF_CALL_float(REGISTER_CPU);
137 TF_CALL_float(DECLARE_GPU);
147 TF_CALL_float(REGISTER_GPU);
159 TF_CALL_float(REGISTER_SYCL);
mkl_identity_op.cc 62 TF_CALL_float(REGISTER_MKL_CPU);
where_op.h 31 TF_CALL_float(m); \
batch_norm_op.cc 174 TF_CALL_float(REGISTER_KERNEL);
194 TF_CALL_float(DECLARE_GPU_SPECS);
206 TF_CALL_float(REGISTER_GPU_KERNEL);
218 TF_CALL_float(REGISTER_KERNEL);
230 TF_CALL_float(REGISTER_KERNEL);
253 TF_CALL_float(DECLARE_GPU_SPECS);
265 TF_CALL_float(REGISTER_GPU_KERNEL);
277 TF_CALL_float(REGISTER_KERNEL);
tile_functor_cpu.cc 59 TF_CALL_float(DEFINE_TYPE);
81 TF_CALL_float(DEFINE_TYPE);
sequence_ops.cc 94 TF_CALL_float(REGISTER_SYCL_KERNEL);
101 TF_CALL_float(REGISTER_CPU_KERNEL);
108 TF_CALL_float(REGISTER_GPU_KERNEL);
172 TF_CALL_float(REGISTER_CPU_KERNEL);
178 TF_CALL_float(REGISTER_GPU_KERNEL);
184 TF_CALL_float(REGISTER_SYCL_KERNEL);
softmax_op.cc 86 TF_CALL_float(REGISTER_CPU);
95 TF_CALL_float(REGISTER_CPU);
diag_op.cc 176 TF_CALL_float(REGISTER_DIAGOP);
189 TF_CALL_float(REGISTER_DIAGPARTOP);
215 TF_CALL_float(REGISTER_DIAGOP_GPU);
238 TF_CALL_float(REGISTER_DIAGPARTOP_GPU);
tile_ops.cc 140 TF_CALL_float(HANDLE_TYPE_NAME);
217 TF_CALL_float(HANDLE_TYPE_NAME_CPU);
231 TF_CALL_float(HANDLE_TYPE_NAME_GPU);
242 TF_CALL_float(HANDLE_TYPE_NAME_SYCL);
324 TF_CALL_float(HANDLE_TYPE_NAME);
486 TF_CALL_float(HANDLE_TYPE_NAME_CPU);
496 TF_CALL_float(HANDLE_TYPE_NAME_GPU);
510 TF_CALL_float(HANDLE_TYPE_NAME_SYCL);
578 TF_CALL_float(REGISTER_GPU);
619 TF_CALL_float(REGISTER_SYCL)
    [all...]
bincount_op_gpu.cu.cc 109 TF_CALL_float(REGISTER_GPU_SPEC);
conditional_accumulator_op.cc 56 TF_CALL_float(REGISTER_KERNELS_CPU);
sparse_conditional_accumulator_op.cc 56 TF_CALL_float(REGISTER_KERNELS_CPU);
multinomial_op.cc 252 TF_CALL_float(REGISTER);
272 TF_CALL_float(REGISTER);
320 TF_CALL_float(REGISTER);
342 TF_CALL_float(REGISTER);
stateful_random_ops.cc 213 TF_CALL_float(REGISTER_CPU);
219 TF_CALL_float(REGISTER_GPU);
stateless_random_ops.cc 198 TF_CALL_float(REGISTER_CPU);
206 TF_CALL_float(REGISTER_GPU);
  /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
62 #define TF_CALL_float(m) m(float)
93 #define TF_CALL_float(m) m(float)
123 #define TF_CALL_float(m) m(float)
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 136 TF_CALL_float(REGISTER);
160 TF_CALL_float(DECLARE_FUNCTOR);
180 TF_CALL_float(REGISTER);
  /external/tensorflow/tensorflow/core/kernels/data/experimental/sql/
sqlite_query_connection.cc 101 TF_CALL_float(DOUBLE_CASE)

Completed in 406 milliseconds

1 2 3