Home
last modified time | relevance | path

Searched refs:OUT_TYPE (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcast_op.h30 #define SPECIALIZE_CAST(DEVICE, OUT_TYPE, IN_OUT) \ argument
32 struct CastFunctor<Device, OUT_TYPE, IN_OUT> { \
34 typename TTypes<OUT_TYPE>::Flat out_tensor, \
39 in_tensor.unaryExpr(LSBZeroSetter<IN_OUT, OUT_TYPE>()) \
40 .template cast<OUT_TYPE>(); \
42 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
46 template struct CastFunctor<DEVICE, OUT_TYPE, IN_OUT>;
48 #define SPECIALIZE_CAST(DEVICE, OUT_TYPE, IN_OUT) \ argument
50 struct CastFunctor<DEVICE, OUT_TYPE, IN_OUT> { \
52 typename TTypes<OUT_TYPE>::Flat out_tensor, \
[all …]
/external/gemmlowp/meta/
Dlegacy_multi_thread_common.h41 template <typename IN_TYPE, typename OUT_TYPE, typename F>
48 OUT_TYPE* result;
53 const TaskRect& task_rect, std::int32_t k, OUT_TYPE* result, in MetaTask()
67 OUT_TYPE* task_result = in Run()
115 template <typename IN_TYPE, typename OUT_TYPE, typename F>
120 OUT_TYPE* result, std::int32_t result_stride, in MultiThreadedMatrixMatrix()
140 tasks.push_back(new internal::MetaTask<IN_TYPE, OUT_TYPE, F>( in MultiThreadedMatrixMatrix()
Dlegacy_multi_thread_gemm.h33 template <typename IN_TYPE, typename OUT_TYPE, typename F>
36 std::int32_t n, std::int32_t k, OUT_TYPE* result, in CacheFriendlyMatrixMatrix()
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt19305 type_list_attr: "OUT_TYPE"
19309 type_list_attr: "OUT_TYPE"
19312 name: "OUT_TYPE"
19341 type_list_attr: "OUT_TYPE"
19345 type_list_attr: "OUT_TYPE"
19348 name: "OUT_TYPE"
19384 type_list_attr: "OUT_TYPE"
19388 type_list_attr: "OUT_TYPE"
19391 name: "OUT_TYPE"
19434 type_list_attr: "OUT_TYPE"
[all …]
Dops_history.v2.pbtxt18962 type_list_attr: "OUT_TYPE"
18966 type_list_attr: "OUT_TYPE"
18969 name: "OUT_TYPE"
18998 type_list_attr: "OUT_TYPE"
19002 type_list_attr: "OUT_TYPE"
19005 name: "OUT_TYPE"
19041 type_list_attr: "OUT_TYPE"
19045 type_list_attr: "OUT_TYPE"
19048 name: "OUT_TYPE"
19091 type_list_attr: "OUT_TYPE"
[all …]
Dops_history.v0.pbtxt9641 type_list_attr: "OUT_TYPE"
9645 type_list_attr: "OUT_TYPE"
9648 name: "OUT_TYPE"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt8899 type_list_attr: "OUT_TYPE"
8903 type_list_attr: "OUT_TYPE"
8906 name: "OUT_TYPE"