Searched refs:SetCpuDevice (Results 1 – 1 of 1) sorted by relevance
67 void SetCpuDevice(OpInfo* op_info) { in SetCpuDevice() function78 SetCpuDevice(&op_context.op_info); in DescribeMatMul()114 SetCpuDevice(&op_context.op_info); in DescribeBatchMatMul()127 SetCpuDevice(&op_context.op_info); in DescribeSparseTensorDenseMatMul()164 SetCpuDevice(&op_context.op_info); in DescribeConvolution()181 SetCpuDevice(&op_context.op_info); in DescribeDepthwiseConv2dNative()204 SetCpuDevice(&op_context.op_info); in DescribeFusedConv2DBiasActivation()245 SetCpuDevice(&op_context.op_info); in DescribeUnaryOp()263 SetCpuDevice(&op_context.op_info); in DescribeBinaryOp()279 SetCpuDevice(&op_context.op_info); in DescribeBiasAdd()[all …]