Searched defs:block_dims (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_block_io.cpp | 223 DSizes<Index, 3> block_dims = tensor_dims; in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
270 DSizes<Index, 4> block_dims = tensor_dims; in test_block_io_copy_using_reordered_dimensions_squeeze() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | trace_listener.h | 50 const BlockDim& block_dims, in LaunchSubmit()
|
| H A D | stream_executor_internal.h | 188 const BlockDim& block_dims, const KernelBase& k, in Launch()
|
| H A D | stream.h | 1624 BlockDim block_dims, in ThenLaunch()
|
| H A D | stream_executor_pimpl.cc | 476 const BlockDim& block_dims, in Launch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| H A D | host_gpu_executor.h | 58 const BlockDim& block_dims, const KernelBase& kernel, in Launch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 299 const BlockDim& block_dims, in Launch()
423 const BlockDim& block_dims) { in VlogOccupancyInfo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | spacetobatch_op.cc | 55 const int block_dims = orig_block_shape.dim_size(0); in SpaceToBatchOpCompute() local
|
| H A D | batchtospace_op.cc | 53 const int block_dims = orig_block_shape.dim_size(0); in BatchToSpaceOpCompute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linear_operator_util.py | 565 def split_arg_into_blocks(block_dims, block_dims_fn, arg, axis=-1): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 442 const BlockDim& block_dims, in Launch()
483 const BlockDim& block_dims) { in VlogOccupancyInfo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executor.h | 64 const BlockDim &block_dims, const KernelBase &kernel, in Launch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1888 std::vector<int64_t> block_dims = in TEST_F() local
1922 std::vector<int64_t> block_dims = RandomDims(1, 3, 0, 5); in TEST_F() local
4201 std::vector<int64_t> block_dims = RandomDims(4, 4, 0, 5); in TEST_F() local
4240 std::vector<int64_t> block_dims = RandomDims(1, 3, 0, 5); in TEST_F() local
|