Home
last modified time | relevance | path

Searched defs:block_dims (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dtrace_listener.h50 const BlockDim& block_dims, in LaunchSubmit()
Dstream_executor_internal.h192 const BlockDim &block_dims, const KernelBase &k, in Launch()
Dstream_executor_pimpl.h867 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
Dstream_executor_pimpl.cc480 const BlockDim &block_dims, in Launch()
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning_utils.py96 def expand_tensor(tensor, block_dims): argument
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h58 const BlockDim &block_dims, const KernelBase &kernel, in Launch()
/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc53 const int block_dims = orig_block_shape.dim_size(0); in BatchToSpaceOpCompute() local
Dspacetobatch_op.cc55 const int block_dims = orig_block_shape.dim_size(0); in SpaceToBatchOpCompute() local
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h66 const BlockDim &block_dims, const KernelBase &kernel, in Launch()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc287 const BlockDim& block_dims, const KernelBase& kernel, in Launch()
421 const BlockDim& block_dims) { in VlogOccupancyInfo()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc421 const BlockDim& block_dims, const KernelBase& kernel, in Launch()
468 const BlockDim& block_dims) { in VlogOccupancyInfo()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1231 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