Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDecodeCSV.pbtxt9 type_list_attr: "OUT_TYPE"
13 type_list_attr: "OUT_TYPE"
16 name: "OUT_TYPE"
45 type_list_attr: "OUT_TYPE"
49 type_list_attr: "OUT_TYPE"
52 name: "OUT_TYPE"
88 type_list_attr: "OUT_TYPE"
92 type_list_attr: "OUT_TYPE"
95 name: "OUT_TYPE"
138 type_list_attr: "OUT_TYPE"
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DDecodeCSV.pbtxt9 type_list_attr: "OUT_TYPE"
13 type_list_attr: "OUT_TYPE"
16 name: "OUT_TYPE"
45 type_list_attr: "OUT_TYPE"
49 type_list_attr: "OUT_TYPE"
52 name: "OUT_TYPE"
88 type_list_attr: "OUT_TYPE"
92 type_list_attr: "OUT_TYPE"
95 name: "OUT_TYPE"
138 type_list_attr: "OUT_TYPE"
[all …]
/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 …]
Dstateless_random_ops.cc268 #define REGISTER_POISSON(RATE_TYPE, OUT_TYPE) \ argument
275 .TypeConstraint<OUT_TYPE>("dtype"), \
276 StatelessRandomPoissonOp<RATE_TYPE, OUT_TYPE>)
/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.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.pbtxt11836 type_list_attr: "OUT_TYPE"
11840 type_list_attr: "OUT_TYPE"
11843 name: "OUT_TYPE"