Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_mkl.cc73 memory::dims conv1_dst_dim = {ToInt(input_batch), ToInt(kernel_filters), in MKLConvImpl() local
95 memory({{{conv1_dst_dim}, memory::data_type::f32, memory::format::nhwc}, in MKLConvImpl()
106 {conv1_dst_dim}, memory::data_type::f32, memory::format::any); in MKLConvImpl()