| /external/tensorflow/tensorflow/core/kernels/ |
| D | batchtospace_op.cc | 53 const int block_dims = orig_block_shape.dim_size(0); in BatchToSpaceOpCompute() local
55 context, orig_input_tensor.dims() >= 1 + block_dims, in BatchToSpaceOpCompute()
56 errors::InvalidArgument("input rank should be >= ", 1 + block_dims, in BatchToSpaceOpCompute()
61 block_dims == orig_crops.dim_size(0) && in BatchToSpaceOpCompute()
63 errors::InvalidArgument("crops should have shape [", block_dims, in BatchToSpaceOpCompute()
76 for (; removed_prefix_block_dims < block_dims; ++removed_prefix_block_dims) { in BatchToSpaceOpCompute()
87 for (; removed_suffix_block_dims < block_dims - removed_prefix_block_dims; in BatchToSpaceOpCompute()
89 const int dim = block_dims - 1 - removed_suffix_block_dims; in BatchToSpaceOpCompute()
98 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in BatchToSpaceOpCompute()
114 block_dims - removed_prefix_block_dims - removed_suffix_block_dims; in BatchToSpaceOpCompute()
[all …]
|
| D | spacetobatch_op.cc | 55 const int block_dims = orig_block_shape.dim_size(0); in SpaceToBatchOpCompute() local
56 if (orig_input_tensor.dims() < 1 + block_dims) { in SpaceToBatchOpCompute()
57 return errors::InvalidArgument("input rank should be >= ", 1 + block_dims, in SpaceToBatchOpCompute()
62 block_dims == orig_paddings.dim_size(0) && in SpaceToBatchOpCompute()
64 return errors::InvalidArgument("paddings should have shape [", block_dims, in SpaceToBatchOpCompute()
79 for (; removed_prefix_block_dims < block_dims; ++removed_prefix_block_dims) { in SpaceToBatchOpCompute()
90 for (; removed_suffix_block_dims < block_dims - removed_prefix_block_dims; in SpaceToBatchOpCompute()
92 const int dim = block_dims - 1 - removed_suffix_block_dims; in SpaceToBatchOpCompute()
101 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in SpaceToBatchOpCompute()
110 block_dims - removed_prefix_block_dims - removed_suffix_block_dims; in SpaceToBatchOpCompute()
[all …]
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning_utils.py | 96 def expand_tensor(tensor, block_dims): argument
127 if len(block_dims) != 2:
130 block_height, block_width = block_dims
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 287 const BlockDim& block_dims, const KernelBase& kernel, in Launch() argument
300 VlogOccupancyInfo(kernel, thread_dims, block_dims); in Launch()
331 GetGpuContext(stream), hipfunc, block_dims.x, block_dims.y, in Launch()
332 block_dims.z, thread_dims.x, thread_dims.y, thread_dims.z, in Launch()
337 << "; block dim: " << block_dims.ToString(); in Launch()
421 const BlockDim& block_dims) { in VlogOccupancyInfo() argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 421 const BlockDim& block_dims, const KernelBase& kernel, in Launch() argument
434 VlogOccupancyInfo(kernel, thread_dims, block_dims); in Launch()
448 if (!GpuDriver::LaunchKernel(context_, cufunc, block_dims.x, block_dims.y, in Launch()
449 block_dims.z, thread_dims.x, thread_dims.y, in Launch()
456 << "; block dim: " << block_dims.ToString(); in Launch()
468 const BlockDim& block_dims) { in VlogOccupancyInfo() argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | trace_listener.h | 50 const BlockDim& block_dims, in LaunchSubmit() argument
|
| D | stream_executor_pimpl.h | 453 const BlockDim &block_dims, const KernelBase &kernel,
867 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch() argument
882 parent_->Launch(this, thread_dims, block_dims, kernel, kernel_args); in ThenLaunch()
|
| D | stream_executor_pimpl.cc | 480 const BlockDim &block_dims, in Launch() argument
483 SubmitTrace(&TraceListener::LaunchSubmit, stream, thread_dims, block_dims, in Launch()
486 return implementation_->Launch(stream, thread_dims, block_dims, kernel, args); in Launch()
|
| D | stream_executor_internal.h | 192 const BlockDim &block_dims, const KernelBase &k, in Launch() argument
|
| D | stream.h | 171 Stream &ThenLaunch(ThreadDim thread_dims, BlockDim block_dims,
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 73 const BlockDim& block_dims, const KernelBase& k,
270 const BlockDim& block_dims);
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1231 std::vector<int64> block_dims = in TEST_F() local
1239 input_dims[1 + i] = block_dims[i]; in TEST_F()
1265 std::vector<int64> block_dims = RandomDims(1, 3, 0, 5); in TEST_F() local
1266 int num_block_dims = block_dims.size(); in TEST_F()
1269 RandomDims(block_dims.size(), block_dims.size(), 0, 4); in TEST_F()
1274 input_dims[0] *= block_dims[i]; in TEST_F()
1297 std::vector<int32>(block_dims.begin(), block_dims.end()))) in TEST_F()
2957 std::vector<int64> block_dims = RandomDims(4, 4, 0, 5); in TEST_F() local
2964 input_dims[1 + i] = block_dims[i] * block_size; in TEST_F()
2996 std::vector<int64> block_dims = RandomDims(1, 3, 0, 5); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 66 const BlockDim &block_dims, const KernelBase &kernel, in Launch() argument
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 58 const BlockDim &block_dims, const KernelBase &kernel, in Launch() argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1630 md.layout_desc.blocking.block_dims[i] = 1; in CreateBlockedMemDescHelper()
|