Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc263 const Shape& operand1_shape = desc.operand1_shape; in GetGpuConvConfig() local
284 config.filter_shape = operand1_shape; in GetGpuConvConfig()
289 config.filter_shape = operand1_shape; in GetGpuConvConfig()
295 config.output_shape = operand1_shape; in GetGpuConvConfig()
448 descriptor.operand1_shape = cudnn_call->operand(1)->shape(); in GetGpuConvConfig()
Dgpu_conv_runner.h129 Shape operand1_shape; member
Dir_emitter_unnested.cc1149 descriptor.operand1_shape = in EmitConvolutionThunkFromMlir()