Searched refs:out_t (Results 1 – 9 of 9) sorted by relevance
/external/mesa3d/src/mesa/drivers/dri/nouveau/ |
D | nouveau_array.c | 33 #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/ |
D | index_ops_kernel_argmax_float_1d.cc | 35 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()
|
D | index_ops_kernel_argmax_float_2d.cc | 39 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/ |
D | sparse_tensor_dense_add_op.cc | 83 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()
|
D | set_kernels.cc | 265 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/ |
D | mpi_ops.cc | 164 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/ |
D | mpi_ops.cc | 165 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/ |
D | sparse_tensor.h | 459 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/ |
D | session_test.py | 1414 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 …]
|