Searched defs:block_dims (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | trace_listener.h | 50 const BlockDim& block_dims, in LaunchSubmit()
|
D | stream_executor_internal.h | 192 const BlockDim &block_dims, const KernelBase &k, in Launch()
|
D | stream_executor_pimpl.h | 867 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
|
D | stream_executor_pimpl.cc | 480 const BlockDim &block_dims, in Launch()
|
/external/tensorflow/tensorflow/contrib/model_pruning/python/ |
D | pruning_utils.py | 96 def expand_tensor(tensor, block_dims): argument
|
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.h | 58 const BlockDim &block_dims, const KernelBase &kernel, in Launch()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | batchtospace_op.cc | 53 const int block_dims = orig_block_shape.dim_size(0); in BatchToSpaceOpCompute() local
|
D | spacetobatch_op.cc | 55 const int block_dims = orig_block_shape.dim_size(0); in SpaceToBatchOpCompute() local
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | executor.h | 66 const BlockDim &block_dims, const KernelBase &kernel, in Launch()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 287 const BlockDim& block_dims, const KernelBase& kernel, in Launch() 421 const BlockDim& block_dims) { in VlogOccupancyInfo()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 421 const BlockDim& block_dims, const KernelBase& kernel, in Launch() 468 const BlockDim& block_dims) { in VlogOccupancyInfo()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 1231 std::vector<int64> block_dims = in TEST_F() local 1265 std::vector<int64> block_dims = RandomDims(1, 3, 0, 5); in TEST_F() local 2957 std::vector<int64> block_dims = RandomDims(4, 4, 0, 5); in TEST_F() local 2996 std::vector<int64> block_dims = RandomDims(1, 3, 0, 5); in TEST_F() local
|