Searched refs:is_multi_device_function (Results 1 – 22 of 22) sorted by relevance
129 bool is_multi_device_function = false; member150 bool is_multi_device_function() const { return is_multi_device_function_; } in is_multi_device_function() function172 is_multi_device_function_(params.is_multi_device_function), in FunctionMetadata()236 bool is_multi_device_function() const { in is_multi_device_function() function237 return metadata_->is_multi_device_function(); in is_multi_device_function()
542 TF_RETURN_IF_ERROR(IsMultiDevice(ctx, &inst_opts.is_multi_device_function)); in Instantiate()548 if (inst_opts.is_multi_device_function) { in Instantiate()847 captured_func_->is_multi_device_function(); in ShouldCreateRendezvous()858 if (!metadata_->is_multi_device_function()) { in IsMultiDevice()
205 params.is_multi_device_function = true; in MapDatasetOp()
240 params.is_multi_device_function = true; in FilterDatasetOp()
258 params.is_multi_device_function = true; in FlatMapDatasetOp()
300 params.is_multi_device_function = true; in InterleaveDatasetOp()
207 params.is_multi_device_function = true; in ParallelMapDatasetOp()
592 params.is_multi_device_function = true; in ReduceDatasetOp()
1429 params.is_multi_device_function = true; in ParallelInterleaveDatasetOp()
37 params.is_multi_device_function = true; in TakeWhileDatasetOp()
37 params.is_multi_device_function = true; in ScanDatasetOp()
36 params.is_multi_device_function = true; in GroupByReducerDatasetOp()
37 params.is_multi_device_function = true; in GroupByWindowDatasetOp()
763 params.is_multi_device_function = true; in MapAndBatchDatasetOp()
1098 params.is_multi_device_function = true; in ParallelInterleaveDatasetOp()
198 opts.is_multi_device_function = true; in Instantiate()
143 options.is_multi_device_function = true; in InstantiateFunc()
539 bool is_multi_device_function = false; member
559 options.is_multi_device_function = true; in TEST_F()
587 inst_opts.is_multi_device_function = true; in MakeOptions()
1082 if (options.is_multi_device_function) { in Instantiate()
713 if (options.is_multi_device_function) return false; in IsLocalTarget()