| /external/tensorflow/tensorflow/core/kernels/ |
| batchtospace_op.cc | 53 const int block_dims = orig_block_shape.dim_size(0); local
55 context, orig_input_tensor.dims() >= 1 + block_dims,
56 errors::InvalidArgument("input rank should be >= ", 1 + block_dims,
61 block_dims == orig_crops.dim_size(0) &&
63 errors::InvalidArgument("crops should have shape [", block_dims,
76 for (; removed_prefix_block_dims < block_dims; ++removed_prefix_block_dims) {
87 for (; removed_suffix_block_dims < block_dims - removed_prefix_block_dims;
89 const int dim = block_dims - 1 - removed_suffix_block_dims;
98 for (int block_dim = 0; block_dim < block_dims; ++block_dim) {
114 block_dims - removed_prefix_block_dims - removed_suffix_block_dims
[all...] |
| spacetobatch_op.cc | 55 const int block_dims = orig_block_shape.dim_size(0); local
57 context, orig_input_tensor.dims() >= 1 + block_dims,
58 errors::InvalidArgument("input rank should be >= ", 1 + block_dims,
63 block_dims == orig_paddings.dim_size(0) &&
66 block_dims, ", 2] instead of ",
79 for (; removed_prefix_block_dims < block_dims; ++removed_prefix_block_dims) {
90 for (; removed_suffix_block_dims < block_dims - removed_prefix_block_dims;
92 const int dim = block_dims - 1 - removed_suffix_block_dims;
101 for (int block_dim = 0; block_dim < block_dims; ++block_dim) {
110 block_dims - removed_prefix_block_dims - removed_suffix_block_dims
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| trace_listener.h | 51 const BlockDim& block_dims,
|
| stream_executor_pimpl.h | 411 const BlockDim &block_dims, const KernelBase &kernel,
[all...] |
| stream_executor_internal.h | 168 const BlockDim &block_dims, const KernelBase &k,
|
| stream_executor_pimpl.cc | 426 const BlockDim &block_dims,
429 SubmitTrace(&TraceListener::LaunchSubmit, stream, thread_dims, block_dims,
432 return implementation_->Launch(stream, thread_dims, block_dims, kernel, args);
|
| stream.h | 158 Stream &ThenLaunch(ThreadDim thread_dims, BlockDim block_dims,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_gpu_executor.cc | 351 const BlockDim &block_dims, const KernelBase &kernel,
364 VlogOccupancyInfo(kernel, thread_dims, block_dims);
378 if (!CUDADriver::LaunchKernel(context_, cufunc, block_dims.x, block_dims.y,
379 block_dims.z, thread_dims.x, thread_dims.y,
386 << "; block dim: " << block_dims.ToString();
398 const BlockDim &block_dims) {
424 int block_count = block_dims.x * block_dims.y * block_dims.z
[all...] |
| cuda_gpu_executor.h | 68 const BlockDim &block_dims, const KernelBase &k,
233 const BlockDim &block_dims);
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| executor.h | 67 const BlockDim &block_dims, const KernelBase &kernel,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| host_gpu_executor.h | 59 const BlockDim &block_dims, const KernelBase &kernel,
|
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | [all...] |
