Searched refs:is_gpu_ (Results 1 – 7 of 7) sorted by relevance
36 is_gpu_(ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) {} in XlaArgMinMaxOp()71 if (is_gpu_) { in Compile()77 if (is_gpu_) { in Compile()
120 is_gpu_(ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) {} in DiagPartOp()152 xla::Reshape(is_gpu_ ? xla::GetMatrixDiagonalViaGather(reshape_input) in Compile()160 const bool is_gpu_; member in tensorflow::__anon4ec11fc90111::DiagPartOp
41 is_gpu_(ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) {} in CategoricalOp()114 if (is_gpu_) { in Compile()145 bool is_gpu_; member in tensorflow::__anon324214720111::CategoricalOp
33 const bool is_gpu_; variable
348 is_gpu_(context->device_type().type_string() == DEVICE_GPU_XLA_JIT) { in MatrixDiagPartOp()404 0, is_gpu_ ? xla::GetMatrixDiagonalViaGather(input, upper_diag_index) in Compile()411 is_gpu_ ? xla::GetMatrixDiagonalViaGather(input, diag_index) in Compile()438 const bool is_gpu_; member in tensorflow::MatrixDiagPartOp
114 bool IsGpu(DeviceId device) const { return is_gpu_[device.id()]; } in IsGpu()159 std::vector<bool> is_gpu_; variable
66 is_gpu_.push_back(device_type->type_string() == DEVICE_GPU); in GetIdFor()