Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc149 const Tensor& execution_config = context->input(1); in DoWork() local
150 TF_RET_CHECK(TensorShapeUtils::IsScalar(execution_config.shape())); in DoWork()
153 config_proto.ParseFromString(execution_config.scalar<string>()())); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.cc218 auto execution_config = in Execute() local
221 execution_config, arguments); in Execute()
236 inputs.insert({execution_config, e.SerializeAsString()}); in Execute()