Searched refs:partition_ids_list (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 425 OpInputList partition_ids_list; in Compute() local
427 context->input_list("partition_ids", &partition_ids_list)); in Compute()
446 [&context, &resource_handle_list, &partition_ids_list, in Compute()
469 partition_ids_list[resource_handle_idx], in Compute()
490 OpInputList partition_ids_list; in Compute() local
492 context->input_list("partition_ids", &partition_ids_list)); in Compute()
511 [&context, &resource_handle_list, &partition_ids_list, in Compute()
534 partition_ids_list[resource_handle_idx], in Compute()
|
| D | training_ops.cc | 365 OpInputList partition_ids_list; in Compute() local
369 context->input_list("partition_ids", &partition_ids_list)); in Compute()
380 FindBestSplitsPerPartitionNormal(context, partition_ids_list, in Compute()
442 OpKernelContext* const context, const OpInputList& partition_ids_list, in FindBestSplitsPerPartitionNormal() argument
448 const auto& partition_ids = partition_ids_list[handler_id].vec<int32>(); in FindBestSplitsPerPartitionNormal()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1056 partition_ids_list = []
1065 partition_ids_list.append(partition_ids)
1072 array_ops.shape_n(partition_ids_list), [len(partition_ids_list)])
1073 partition_ids = array_ops.concat(partition_ids_list, axis=0)
1099 partition_ids_list = list(array_ops.split(partition_ids, sizes, axis=0))
1113 partition_ids=partition_ids_list,
|