Searched refs:SetCpuDevice (Results 1 – 1 of 1) sorted by relevance
77 void SetCpuDevice(OpInfo* op_info) { in SetCpuDevice() function88 SetCpuDevice(&op_context.op_info); in DescribeMatMul()125 SetCpuDevice(&op_context.op_info); in DescribeSparseTensorDenseMatMul()141 SetCpuDevice(&op_context.op_info); in DescribeXlaEinsum()205 SetCpuDevice(&op_context.op_info); in DescribeConvolution()222 SetCpuDevice(&op_context.op_info); in DescribeDepthwiseConv2dNative()249 SetCpuDevice(&op_context.op_info); in DescribeFusedConv2DBiasActivation()305 SetCpuDevice(&op_context.op_info); in DescribeUnaryOp()323 SetCpuDevice(&op_context.op_info); in DescribeBinaryOp()339 SetCpuDevice(&op_context.op_info); in DescribeBiasAdd()[all …]