Searched refs:group_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 392 ShapeArray* group_shape) { in GroupShapeFromInputs() argument
398 *group_shape = group_shape_1; in GroupShapeFromInputs()
403 void PopulateGroupIndices(const int64 flat_group_index, VarDimArray group_shape, in PopulateGroupIndices() argument
407 for (int group_dim_index = group_shape.size() - 1; group_dim_index >= 0; in PopulateGroupIndices()
409 const auto group_dim = group_shape[group_dim_index]; in PopulateGroupIndices()
433 ShapeArray group_shape; in ComputeDenseToDense() local
436 OP_REQUIRES_OK(ctx, GroupShapeFromInputs(shape1, shape2, &group_shape)); in ComputeDenseToDense()
450 TensorShapeUtils::NumElements(group_shape, &num_elements)); in ComputeDenseToDense()
453 PopulateGroupIndices(flat_group_index, group_shape, &group_indices); in ComputeDenseToDense()
472 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(group_shape, &output_shape)); in ComputeDenseToDense()
[all …]
|