Home
last modified time | relevance | path

Searched refs:dims_in_elems (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc119 absl::Span<const int64> dims_in_elems, int64 tile_size_y, int64 tile_size_x, in KernelMappingScheme() argument
123 dims_in_elems_(dims_in_elems.begin(), dims_in_elems.end()), in KernelMappingScheme()
Dkernel_tiling.h98 KernelMappingScheme(absl::Span<const int64> dims_in_elems, int64 tile_size_y,