Searched refs:sizes_vec (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | mpi_ops.cc | 158 std::vector<size_t> sizes_vec; member
427 std::vector<size_t> sizes_vec; in PerformCollectiveOp() local
448 sizes_vec = record.sizes_vec; in PerformCollectiveOp()
472 context, input_tensor, sizes_vec, output_tensor) in PerformCollectiveOp()
474 context, input_tensor, sizes_vec, output_tensor); in PerformCollectiveOp()
477 context, input_tensor, sizes_vec, output_tensor) in PerformCollectiveOp()
479 sizes_vec, output_tensor); in PerformCollectiveOp()
482 context, input_tensor, sizes_vec, output_tensor) in PerformCollectiveOp()
484 context, input_tensor, sizes_vec, output_tensor); in PerformCollectiveOp()
1063 record.sizes_vec.push_back(1); in ComputeAsync()
[all …]
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops.cc | 159 std::vector<size_t> sizes_vec; member
428 std::vector<size_t> sizes_vec; in PerformCollectiveOp() local
449 sizes_vec = record.sizes_vec; in PerformCollectiveOp()
473 context, input_tensor, sizes_vec, output_tensor) in PerformCollectiveOp()
475 context, input_tensor, sizes_vec, output_tensor); in PerformCollectiveOp()
478 context, input_tensor, sizes_vec, output_tensor) in PerformCollectiveOp()
480 sizes_vec, output_tensor); in PerformCollectiveOp()
483 context, input_tensor, sizes_vec, output_tensor) in PerformCollectiveOp()
485 context, input_tensor, sizes_vec, output_tensor); in PerformCollectiveOp()
1139 record.sizes_vec.push_back(1); in ComputeAsync()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1395 auto sizes_vec = sizes_value->vec<T>(); in SliceHelper() local
1398 if (sizes_vec(i) != -1) { in SliceHelper()
1400 if (sizes_vec(i) < 0) { in SliceHelper()
1403 ": sizes vector cannot be < -1, but was ", sizes_vec(i)); in SliceHelper()
1406 dims->emplace_back(c->MakeDim(sizes_vec(i))); in SliceHelper()
|