| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_kernel_argmax_float_1d.cc | 38 int64* out_t = static_cast<int64*>(out); local
39 TTypes<int64, 0>::Tensor out_eig(out_t, out_eig_sizes);
|
| index_ops_kernel_argmax_float_2d.cc | 39 int64* out_t = static_cast<int64*>(out); local
41 TTypes<int64, 1>::Tensor out_eig(out_t, out_eig_sizes);
|
| /external/mesa3d/src/mesa/drivers/dri/nouveau/ |
| nouveau_array.c | 33 #define EXTRACT(in_t, out_t) extract_func_##in_t##_to_##out_t
35 #define EXTRACT_FUNC(in_t, out_t, k) \
36 static out_t EXTRACT(in_t, out_t) \
40 return (out_t)x / (k); \
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_ops.cc | 95 Tensor* out_t; variable
96 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, in_t->shape(), &out_t), done);
108 compute_stream, in_t, out_t, std::move(actual_done)); variable
149 Tensor* out_t; variable
150 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, in_t.shape(), &out_t), done);
162 compute_stream, &in_t, out_t, std::move(actual_done)); variable
209 Tensor* out_t; variable
210 OP_REQUIRES_OK_ASYNC(c, c->allocate_output(0, shape, &out_t), done);
221 gpu_info->gpu_id, gpu_info->event_mgr, compute_stream, out_t, variable
|
| nccl_manager.h | 45 // receiving the result of the all-reduce in <out_t>. The device for this
49 // This is an asynchronous call. When <done_callback> is called, <out_t> has
62 const Tensor* in_t, Tensor* out_t,
76 Tensor* out_t, DoneCallback done_callback);
91 const Tensor* in_t, Tensor* out_t,
|
| nccl_manager.cc | 95 Participant(const Tensor* in_t, Tensor* out_t, EventMgr* event_mgr,
100 out_t(out_t),
116 Tensor* out_t; member in struct:tensorflow::NcclManager::Participant
304 const Tensor* in_t, Tensor* out_t,
307 new Participant(in_t, out_t, event_mgr, tensor_stream, executor,
319 new Participant(in_t, nullptr /* out_t */, event_mgr, tensor_stream,
330 Tensor* out_t, DoneCallback done_callback) {
332 new Participant(nullptr /* in_t */, out_t, event_mgr, tensor_stream,
334 AddParticipant(num_devices, key, std::move(participant), out_t->dtype()
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_tensor_dense_add_op.cc | 83 Tensor *out_t; variable
84 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, b->shape(), &out_t));
93 auto out_tensor = out_t->tensor<T, N>(); \
|
| set_kernels.cc | 268 Tensor* out_t; local
269 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, output_shape_ts, &out_t));
270 auto out = out_t->flat<int32>();
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| mpi_ops.cc | 164 Tensor* out_t; member in struct:tensorflow::contrib::mpi_collectives::__anon39405::CollectiveOpRecord
450 output_tensor = record.out_t;
[all...] |
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor.h | 382 auto out_t = out->flat<T>(); local
383 out_t.setConstant(T());
393 auto out_t = out->flat<T>();
417 out_t(ix) = vals_t(n);
|
| /external/tensorflow/tensorflow/python/client/ |
| session_test.py | [all...] |
