Home
last modified time | relevance | path

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

/external/mesa3d/src/mesa/drivers/dri/nouveau/
Dnouveau_array.c33 #define EXTRACT(in_t, out_t) extract_func_##in_t##_to_##out_t argument
35 #define EXTRACT_FUNC(in_t, out_t, k) \ argument
36 static out_t EXTRACT(in_t, out_t) \
40 return (out_t)x / (k); \
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops_kernel_argmax_float_1d.cc35 int64* out_t = static_cast<int64*>(out); in argmax_float_1d_xla_impl() local
36 TTypes<int64, 0>::Tensor out_eig(out_t, out_eig_sizes); in argmax_float_1d_xla_impl()
Dindex_ops_kernel_argmax_float_2d.cc39 int64* out_t = static_cast<int64*>(out); in argmax_float_2d_xla_impl() local
41 TTypes<int64, 1>::Tensor out_eig(out_t, out_eig_sizes); in argmax_float_2d_xla_impl()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensor_dense_add_op.cc83 Tensor *out_t; in Compute() local
84 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, b->shape(), &out_t)); in Compute()
93 auto out_tensor = out_t->tensor<T, N>(); \ in Compute()
Dset_kernels.cc265 Tensor* out_t; in Compute() local
266 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, output_shape_ts, &out_t)); in Compute()
267 auto out = out_t->flat<int32>(); in Compute()
/external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
Dmpi_ops.cc164 Tensor* out_t; member
450 output_tensor = record.out_t; in PerformCollectiveOp()
984 record.out_t = output_tensor; in ComputeAsync()
1088 record.out_t = output_tensor; in ComputeAsync()
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_ops.cc165 Tensor* out_t; member
451 output_tensor = record.out_t; in PerformCollectiveOp()
1038 record.out_t = output_tensor; in ComputeAsync()
1164 record.out_t = output_tensor; in ComputeAsync()
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h459 auto out_t = out->flat<T>(); in ValidateAndInitializeToDense() local
460 out_t.setConstant(T()); in ValidateAndInitializeToDense()
470 auto out_t = out->flat<T>(); in ToDense() local
494 out_t(ix) = vals_t(n); in ToDense()
/external/tensorflow/tensorflow/python/client/
Dsession_test.py1414 out_t = array_ops.identity(feed_t)
1428 sess.run(out_t, feed_dict={
1438 [out_t, feed_t], feed_dict={
1444 feed_fetch_runner = sess.make_callable([out_t, feed_t], [feed_t])
1521 out_t = array_ops.identity(feed_t)
1524 sess.run(out_t, feed_dict={feed_t: feed_val})
1526 out_t.eval(feed_dict={feed_t: feed_val})
1528 out_t.op.run(feed_dict={feed_t: feed_val})
1537 out_t = constant_op.constant(1.0)
1541 sess.run(out_t, feed_dict={feed_int_implicit_int32: largest_int64})
[all …]