Searched refs:use_default_device_ (Results 1 – 2 of 2) sorted by relevance
39 ctx->GetAttr("use_default_device", &use_default_device_)); in ScanDatasetOp()40 params.use_default_device = use_default_device_; in ScanDatasetOp()67 output_shapes_, preserve_cardinality_, use_default_device_); in MakeDataset()88 use_default_device_(use_default_device) { in Dataset()154 b->BuildAttrValue(use_default_device_, &use_default_device_attr); in AsGraphDefInternal()311 const bool use_default_device_; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp::Dataset319 bool use_default_device_; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp
104 bool use_default_device() const { return use_default_device_; } in use_default_device()116 use_default_device_(params.use_default_device), in FunctionMetadata()122 bool use_default_device_ = true; variable