Searched refs:component_index_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | barrier_ops.cc | 528 context->GetAttr("component_index", &component_index_)); in InsertManyOp()
535 ctx, component_index_ < barrier->num_components(), in ComputeAsync()
537 component_index_, " > num_components", in ComputeAsync()
543 barrier->component_type(component_index_)}, in ComputeAsync()
551 barrier->TryInsertMany<T>(*keys, component_index_, *values, ctx, callback); in ComputeAsync()
555 int component_index_; member in tensorflow::barrier::InsertManyOp
|