Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:tile_size_m (Results 1 - 4 of 4) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_options.cc	85 int64 tile_size_m; local 89 CHECK(absl::SimpleAtoi(tile_components[0], &tile_size_m)); 98 return std::tuple<int64, int64, int64>(tile_size_m, tile_size_k,
tiled_dot_emitter.h	48 tensorflow::int64 tile_size_m, tensorflow::int64 tile_size_k,
tiled_dot_emitter.cc	652 // [`tile_size_m`, `tile_size_k`] from the LHS and [`tile_size_k`, 658 int64 min_vectorization_width, int64 tile_size_m, 665 tile_size_m_(tile_size_m), 671 "_", min_vectorization_width(), "_", tile_size_m(), 681 int64 tile_size_m() const { return tile_size_m_; } function in class:xla::cpu::__anon44367::TiledSmallGemmEmitter::Config 734 int64 tile_size_m, llvm::Value* m_start, 749 int64 tile_size_m() const { return config().tile_size_m(); } function in class:xla::cpu::__anon44367::TiledSmallGemmEmitter 823 const int64 m_end = dims().m() - dims().m() % tile_size_m(); 824 EmitTiledGemm(vsl, tile_size_k, k_start, k_end, n_start, n_end, tile_size_m(),     [all...]
dot_op_emitter.cc	249 int64 tile_size_m, tile_size_k, tile_size_n_in_vector_width; local 250 std::tie(tile_size_m, tile_size_k, tile_size_n_in_vector_width) = 259 /*tile_size_m=*/tile_size_m, /*tile_size_k=*/tile_size_k, /*lhs=*/lhs,     [all...]

Completed in 2749 milliseconds